清华源安装 pytorch

开启清华源。重要的是https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 这个源一定要添加进去，然后才能正常安装pytorch
这是conda源,国内好像也就只有清华这么一家conda源.

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ 
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

使用命令 nvcc --version 可以查看cuda的版本

然后到这里去生成安装需要的命令 https://pytorch.org/get-started/locally/

安装GPU版的如下. 我的cuda版本是10.2所以我的命令是下面的

conda install pytorch torchvision cudatoolkit=10.2 # -c pytorch 这个要去掉

然后执行命令 注意删掉最后的参数 -c pytorch
如果要安装无GPU的版本的命令如下

conda install pytorch torchvision cpuonly # -c pytorch 这个要去掉

如果实在还行下载不到, 再请出我们的下载大神迅雷.
然后用浏览器打开这个网址,
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
去里面找合适自己的包
例如我找到的

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/pytorch-1.6.0-py3.8_cuda102_cudnn7_0.tar.bz2

然后用迅雷下载, 速度肯定贼快. 下载完放到C:\ProgramData\Anaconda3\pkgs目录再执行本地安装就快多了. 话说为什么anconda不用这种迅雷的下载加速功能呢…
具体你的pkgs目录是在哪里? 可以通过在执行命令的时候按下Ctrl+C时弹出的提示显示出来

conda install --use-local F:\ProgramData\Anaconda3\pkgs\pytorch-1.6.0-py3.8_cuda102_cudnn7_0.tar.bz2

下面是pip的安装方式

经过实践操作, 我发现, 其实pip比conda更好用一些… 因为pip的 whl安装文件可以直接用迅雷下载, 然后用pip命令就可以直接安装,非常方便. 而conda日后升级会神经质的一定要去再下载一次.

pip 默认清华源的方法 注意这是pip源, 不是Conda源
清华：https://pypi.tuna.tsinghua.edu.cn/simple
阿里云：http://mirrors.aliyun.com/pypi/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
华中理工大学：http://pypi.hustunique.com/
山东理工大学：http://pypi.sdutlinux.org/
豆瓣：http://pypi.douban.com/simple/

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

如果你是arm平台, 那么建议使用 -f https://torch.kmtea.eu/whl/stable-cn.html 参数会很快

另外如果你的平台是arm平台, 建议使用下面的源, 默认都会去加载arm的包.

pip config set global.index-url https://www.piwheels.org/simple

首先还是去 https://pytorch.org/get-started/locally/ 生成安装命令.
选择pip

把命令复制下来到cmd中运行, 一般弹出第一个要下载的whl文件地址就是pytorch 的安装文件. 由于比较大, 建议终止安装,把url复制出来, 用迅雷下载… 然后放到某个英文目录下. 例如F:\whl\torch-1.8.1+cu102-cp38-cp38-win_amd64.whl

然后使用命令单独安装torch, 会很快成功的.

$ pip3 install F:\whl\torch-1.8.1+cu102-cp38-cp38-win_amd64.whl

然后再安装另外两个项目, 把命令中的torch去掉

$ pip3 install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

改成

$ pip3 install torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

安装完可用下面的代码看看安装效果.

import torch
cc = torch.cuda.is_available()
print("torch是否支持GPU:"+ str(cc))
print("torch版本:"+ torch.__version__) 

希望能帮到你.