pytorch实战之Win10下安装pytorch1.2及cuda10.0

环境

Win10+Anaconda3+python3.7

安装cuda10.0

下载cuda10.0

选择window、local版本下载，下载完成后按照默认提示安装，安装前貌似必须关闭掉nvidia显卡(因为我开着网游安装时，在兼容性检查时一直提示缺失部分文件，关闭游戏及显卡后安装就成功了)。更正：把杀毒软件关掉(来自评论栏网友)。

 

验证cuda

cuda默认安装路径：C:\Program Files\NVIDIA GPU Computing Toolkit\cuda\v10.0\bin

window+R进入cmd命令行查看cuda版本：

nvcc -V

        

安装pytorch1.2

配置国内源(提高安装速度)：

# 配置国内源，方便安装Numpy,Matplotlib等
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/
# 配置国内源，安装PyTorch用
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# 显示源地址
conda config --set show_channel_urls yes

创建虚拟环境：

conda create -n pytorchEnv python==3.7  #环境名:pytorchEnv,py版本3.7

进入虚拟环境pytorchEnv：

activate pytorchEnv

安装pytorch1.2：

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0  #无需手动安装cudnn，会在anaconda中自动下载安装。

验证torch可用：

python   #命令行输入python交互
import torch   #导入包
torch.cuda.is_available()

---

BUG：安装pytorch时错误

CondaVerificationError: The package for cudatoolkit located at D:\Anaconda3\pkgs\cudatoolkit-10.0.130-0 appears to be corrupted. The path 'Library/bin/nvrtc64_100_0.dll' specified in the package manifest cannot be found.

解决：

conda clean --packages --tarballs  
#再重新执行pytorch安装命令，有时网络原因出错，多尝试~
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0