记录贴,面向小白。方案一使用清华源下载,方案二使用whl文件下载。推荐直接看方案二,可以安装指定版本,下载速度也很快。
方案一
首先,打开Anaconda Prompt,
输入conda env list
或conda info -e
, 查看当前所有的隔离环境。
找到你想安装pytorch的环境,激活,我这里选择的是Python3.7,所以activate py37
。
这时,前面括号里的base就切换到了py37,说明环境已激活成功。
若你想新建一个环境来安装pytorch,那么输入
conda create --name {env_name} python={python_version}
例: conda create --name py37 python=3.7
这样新环境就创建成功啦。接下来进入正题。
去pytorch官网看一下,
若是直接输入上述官网给的命令,估计速度太慢,还有可能会连接中断。所以,我这里采用的是conda清华源镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
然后,输入
conda install pytorch-cpu torchvision-cpu -c pytorch
有博主说,把最后的-c pytorch去掉会下的快一点,但我是没去下得更快,大家可以自己试一下。
然后就直接安装,也没什么了。
中间出了两次错误:
Error 1
CondaError: Downloaded bytes did not match Content-Length
url: https://repo.anaconda.com/pkgs/main/linux-64/cudnn-7.0.5-cuda8.0_0.tar.bz2
target_path: /home/yyf/miniconda3/pkgs/cudnn-7.0.5-cuda8.0_0.tar.bz2
Content-Length: 261398285
downloaded bytes: 47463195
在用conda安装包的时候下载包的长度不够导致安装包不成功。原因一般是在下载的时候速度较慢,导致下载timeout而终止。对,用清华源镜像也出现了这个问题,简单一点的办法就是,设置conda下载的timeout,然后再尝试。
conda config --set remote_read_timeout_secs 1000.0
若是不想这样的,可以有其他的方式。
Error 2
UserWarning: mkl-service package failed to import, therefore Intel(R) MKL initialization ensuring it
先去自己环境里看一下,发现是有mkl-service,但是没导进来,这个是咱的锅。我之前直接安装的是python编译器,后来才安的Anaconda. 结果系统环境变量没有改过来。
- 更改环境变量
右击此电脑,选择属性(或通过控制面板,进入‘控制面板\系统和安全\系统’)
进入高级系统设置,点击环境变量,在系统变量中找到Path,编辑,在其中添加
D:\Anaconda3\Library\bin
D:\Anaconda3\Scripts
D:\Anaconda3
我的Anaconda包是放在了D盘,大家照这个格式把自己anaconda的位置写进去就行了。
修改后,一路保存,否则可能修改不成功。然后anaconda的环境变量就配好啦。
方案二
torch和torchvision版本对应关系
torch | torchvision | python |
---|---|---|
1.5.1 | 0.6.1 | >=3.6 |
1.5.0 | 0.6.0 | >=3.6 |
1.4.0 | 0.5.0 | ==2.7, >=3.5, <=3.8 |
1.3.1 | 0.4.2 | ==2.7, >=3.5, <=3.7 |
1.3.0 | 0.4.1 | ==2.7, >=3.5, <=3.7 |
1.2.0 | 0.4.0 | ==2.7, >=3.5, <=3.7 |
1.1.0 | 0.3.0 | ==2.7, >=3.5, <=3.7 |
<1.0.1 | 0.2.2 | ==2.7, >=3.5, <=3.7 |
https://download.pytorch.org/whl/torch_stable.html
这种方式真的超快,超方便。
这个方法是我在重装pytorch时找到的。因为第一次装的是1.1版本,太老了,没有Transformer模型。只能卸了重安。1.2版本就有Transformer模型了。
以torch-1.3示例,先到上面那个网站把你需要的torch和torchvision轮子下载下来。
cp37就是python3.7的意思,后面跟着的是系统类型,以此类推,选择自己需要的便好。
进入anaconda,激活你要安装的那个环境,然后改变路径到你放置torch和torchvision的地方,我放在了d盘
输入
pip install torch-1.3.0+cpu-cp37-cp37m-win_amd64.whl
pip install torchvision-0.4.1+cpu-cp37-cp37m-win_amd64.whl
等待安装成功就行了。
测试
导入没问题,就是安装成功啦。