今天本想体验下最新版的pytorch1.01,结果发现conda install时候HTTP error了,查了一下原来从4月25日其国内的清华源、中科大源等陆续关停服务,无奈又只能恢复到Anaconda官网的原始源,我的心情拔凉拔凉的,只能这样啦,安装过程中如何提速记录分享一下。我的GPU是Nvidia 2080ti,CUDA版本10.0,所以我在不同的虚环境中同时安装CPU版本和GPU版本。
1.恢复官方源
由于清华源和中科大源已经不能使用,因此先移除
conda config --remove-key channels
原始C:\Users\Administrator\.condarc下文件为:
channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
- anaconda-fusion
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
show_channel_urls: true
ssl_verify: true
删除后为:
show_channel_urls: true
ssl_verify: true
2. “想办法”访问pytorch官网pytorch.org,选择平台会生成安装指令
从官网上可以看到,GPU版本的安装指令为:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
CPU版本的安装指令为:
conda install pytorch-cpu torchvision-cpu -c pytorch
注意,这一步只是告诉大家用conda安装时候的命令,实际不能“想办法”上官网的童鞋也可以直接用这个命令装,但不管能不能看到官网,安装pytorch都巨慢,这里使用这个命令只是为了让Conda自动解决安装pytorch时候的依赖,不然自己找来安装有点麻烦,实践证明,除了安装pytorch会超时,其他都很快,不需要“想办法”。我安装时候的顺带自动安装的包如下图:
3.使用IDM Downloader直接下载安装包:
GPU版本:
https://conda.anaconda.org/pytorch/win-64/pytorch-1.0.1-py3.6_cuda100_cudnn7_1.tar.bz2
CPU版本:
https://conda.anaconda.org/pytorch/win-64/pytorch-cpu-1.0.1-py3.6_cpu_1.tar.bz2
用IDM下载巨快,否则巨慢
4.清除本地下载缓存
本地由于第三步conda下载失败,需要先把不完整的下载缓存清除,位置在anaconda/dpkg目录下,刚才未完成下载的pytorch-1.0.1-py3.6_cuda100_cudnn7_1.tar.bz2包及其解压文件夹一起删除。
5.使用第三步下载的安装包本地安装
conda install --use-local pytorch-1.0.1-py3.6_cuda100_cudnn7_1.tar.bz2
conda install --use-local pytorch-cpu-1.0.1-py3.6_cpu_1.tar.bz2
6.测试一下OK
import torch
print(torch.__version__)
7.附上pip包下载地址,当然这个就需要能“想办法”出去下载
https://download.pytorch.org/whl/cpu/torch-1.0.1-cp36-cp36m-win_amd64.whl
https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-win_amd64.whl #cuda10.0
以上步骤中,第三四五步都可以直接进行,不需要“想办法”出去。