Pytorch安装记录

环境

• Win10
• NVIDIA驱动：442.50
• conda 4.8.3

1. 安装CUDA

显卡，CUDA, cudann和Pytorch的对应关系参考这里
显卡驱动版本与CUDA版本的对应关系：

本机显卡驱动对应的CUDA版本是CUDA10.2，去官网下载安装包，然后运行安装，最后用nvcc -V命令查看是否安装成功：

2. 安装CUDANN

去英伟达官网，注册英伟达用户，填写用户问卷，下载与自己CUDA版本匹配的CUDANN：

3. 安装pytorch

创建一个anaconda环境，命名为pytorch，这里必须要指定Python的版本，否则默认安装最高版本，后续可能会出错：

conda create -n pytorch python=3.7

激活环境：

conda activate pytorch

进入pytorch官网，选择与自己OS和CUDA版本相符合的安装命令，这里一定要使用pip命令而不是conda命令，原因是参考了这个博客里的评论，使用的命令如下：

pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

等待安装结束，然后使用以下命令测试是否安装成功：

import torch
print(torch.cuda.is_available)

这里出现了报错：

Microsoft Visual C++ Redistributable is not installed, this may lead to the DLL load failure.
                 It can be downloaded at https://aka.ms/vs/16/release/vc_redist.x64.exe
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "G:\software\anaconda\envs\pytorch\lib\site-packages\torch\__init__.py", line 127, in <module> raise err
OSError: [WinError 126] 找不到指定的模块。 Error loading "G:\software\anaconda\envs\pytorch\lib\site-packages\torch\lib\asmjit.dll" or one of its dependencies.

然后根据上面提示的网址安装vc_redist，重新import就结束了。