针对windows+python3.7环境下安装gpu版Torch出现的问题

首先需要安装cuda和cudnn。需要针对自己的gpu版本安装

查看GPU版本需要进入C:\Program Files\NVIDIA Corporation\NVSMI输入

nvidia-smi

 其中Driver Version就是GPU版本

这是对应的表格

Release Notes :: CUDA Toolkit DocumentationThe Release Notes for the CUDA Toolkit.https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

cuda的下载地址（我是开着梯子去的）

CUDA Toolkit Archive | NVIDIA Developerhttps://developer.nvidia.com/cuda-toolkit-archivecuda安装完后可以通过这种方式检查是否安装成功

nvcc --version

cuDNN的下载地址（也是开着梯子）

https://developer.nvidia.com/rdp/cudnn-downloadhttps://developer.nvidia.com/rdp/cudnn-downloadcuDNN下载解压后将以下三个文件复制到cuda的目录下即可

检验cuDNN安装是否是成功，在命令行中进入“ cuda安装的位置\extras\demo_suite ”，然后输入

bandwidthTest.exe

 

得到Result = PASS则可视为cuDNN安装成功

最后需要安装pytorch

如果开着梯子可以直接运行，应该是能下载到GPU版的pytorch

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

如果配置清华源的话他只能自动匹配下载cpu版的pytorch

所以没有梯子的话可以用这条代码

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

最后可以通过以下命令来验证安装

如果意外print(torch.cuda.is_available())返回false的话

print(torch.cuda.is_available())

可以在NVIDIA控制面板

 再试试print(torch.cuda.is_available())如果但是返回false就重启再试试

还是false的话，就看看别的blog吧（doge）