conda安装pytorch遇到的坑

一、根据cuda选择自己合适的pytorch版本

https://pytorch.org/get-started/locally/

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

二、配置清华镜像

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 config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

三、通过conda安装pytorch

(此处需要删除 -c pytorch，如果不删除，则默认会从pytorch官网下载)

conda install pytorch torchvision cudatoolkit=10.2

四、坑

很多人到上面就结束了，但是我下载的时候有四个包因为网络的原因报错，分别是torchvision、mkl，pytorch，cudatoolkit。
这时候，找到报错包的网址，复制到迅雷（浏览器也下载不下来），下载下来

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/torchvision-0.7.0-py37_cu102.tar.bz2
https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/mkl-2020.2-256.conda
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/pytorch-1.6.0-py3.7_cuda102_cudnn7_0.tar.bz2
https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/cudatoolkit-10.2.89-h74a9793_1.conda

然后找到conda安装包的位置，将下载下来的包复制进去（我的位置是D:\Anaconda3\pkgs）

重点来了，复制进去之后，需要找到目录中的urls.txt文件，将上面的网址复制进去，否则安装的时候会重新从镜像上下载！！！！

弄好之后，通过conda install pytorch torchvision cudatoolkit=10.2安装pytorch，成功！！！

此处参考的这位大佬的解决方法：https://zhuanlan.zhihu.com/p/73741240

五、第二个坑

安装完pytorch 1.6之后，我通过
import torch
导入的时候报错：

OSError: [WinError 126] The specified module could not be found. Error loading "D:\anaconda3\lib\site-packages\torch\lib\asmjit.dll" or one of its dependencies.

瞬间崩溃，我还以为我前面的哪个步骤有问题！！

经过耐心的搜索，最后在STACK OVERFLOW找到了答案

原因是电脑少了一个Microsoft Visual C++的一个依赖环境，通过回答中给的链接
https://aka.ms/vs/16/release/vc_redist.x64.exe
下载、安装、重新测试

import torch
x = torch.rand(5, 3)
print(x)
print(torch.cuda.is_available())    # True

参考：
https://stackoverflow.com/questions/63187161/error-while-import-pytorch-module-the-specified-module-could-not-be-found