解决anaconda安装pytorch报错找不到包的问题

发布时间:

使用anaconda 命令安装pytorch包的问题,以下给出解决办法。

解决anaconda安装pytorch报错找不到包的问题

状况描述:

由于是第一次接触安装,看了很多篇文章,也遇到了一些坑,在此总结一下。

安装pytorch,从硬件上分为2个版本,一个是cpu,一个是gpu(显卡)。gpu版本可以加速训练效果。

由于本人的电脑比较旧,显卡是rtx1050,我安装了cuda9.0版本+cudnn+pytorch(torch+torchvision)

本人安装的是gpu版本的,在这个过程中,走了一些弯路。大致流程是这样:

  • 查看显卡驱动,看支持什么 版本的cuda
  • 下载安装cuda,将cudnn解压到cuda安装目录下
  • 安装 anaconda,创建对应的python版本环境
  • 将pytorch 安装进anaconda对应的环境中

总结:上述的具体流程可以参考其他博文,这里就大致说一下。

问题描述

本人按照上述步骤走到最后一步发现报错,报错如下:

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
Solving environment: failed

PackagesNotFoundError: The following packages are not available from current channels:

- pytorch==1.1.0
- torchvision==0.3.0

Current channels:

- https://conda.anaconda.org/pytorch/win-64
- https://conda.anaconda.org/pytorch/noarch
- https://repo.anaconda.com/pkgs/main/win-64
- https://repo.anaconda.com/pkgs/main/noarch
- https://repo.anaconda.com/pkgs/free/win-64
- https://repo.anaconda.com/pkgs/free/noarch
- https://repo.anaconda.com/pkgs/r/win-64
- https://repo.anaconda.com/pkgs/r/noarch
- https://repo.anaconda.com/pkgs/pro/win-64
- https://repo.anaconda.com/pkgs/pro/noarch
- https://repo.anaconda.com/pkgs/msys2/win-64
- https://repo.anaconda.com/pkgs/msys2/noarch

To search for alternate channels that may provide the conda package you're
looking for, navigate to

https://anaconda.org

and use the search bar at the top of the page.

访问官网,发现官网上面的历史版本最低都是 1.4在往下就没有了。

解决办法:

下载安装包,本地 pip install。

pytorch下载链接

版本号对应查询

假设:电脑已经下载好了 安装包 (torch+torchvision)

打开 Anaconda Prompt

切换到虚拟环境:activate pytorch (此处的pytorch是anaconda 新创建的)

切换盘符到 安装包目录下,使用命令pip install xxx.whl即可

举例如下:

(pytorch) D:\Anaconda3\envs\pytorch\Lib\site-packages\wheel>pip install torch-1.1.0-cp37-cp37m-win_amd64.whl
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Processing d:\anaconda3\envs\pytorch\lib\site-packages\wheel\torch-1.1.0-cp37-cp37m-win_amd64.whl
Collecting numpy
  Using cached https://pypi.tuna.tsinghua.edu.cn/packages/97/9f/da37cc4a188a1d5d203d65ab28d6504e17594b5342e0c1dc5610ee6f4535/numpy-1.21.6-cp37-cp37m-win_amd64.whl (14.0 MB)
Installing collected packages: numpy, torch
Successfully installed numpy-1.21.6 torch-1.1.0