win10 安装CUDA 11.0后再安装GPU版torch的踩坑记录

目录

安装 cuda11 （坑）

卸载cuda_V11

安装cuda_V10

安装cuDnn 

---

cuda历史版本下载地址   https://developer.nvidia.com/cuda-toolkit-archive

cudnn 历史版本下载地址   https://developer.nvidia.com/rdp/cudnn-archive

 

———————— 2020. 11. 02 更新 ————————

感谢评论区的 Mr.Hawk 提醒：

pytorch官网 [ https://pytorch.org/get-started/locally/]  已经又支持CUDA11.0的版本了。

 

            ———————— 2020. 10. 15 更新 ————————

由 原创的cuda11踩坑文（在这儿哈  https://blog.csdn.net/q1042960848/article/details/107856596）评论区，博友帐下幕僚指出，可通过conda install指令的形式安装 cuda11&cudnn8.0的pytorch nightly版本，指令如下：

conda install pytorch torchvision cudatoolkit=11 -c pytorch-nightly

• 作者在conda内测试过，确实可行，只是还没有实际安装不知道有没有什么坑要走。等有时间了再测试一下，在此处更新。

安装 cuda11 （坑）

电脑先安装的cuda11.0 ， 随便搜索下 cuda download,就会给您跳转到这个页面：

https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

然后根据网上的教程（为避免报错，不安装 visual studio integration；） Win10自定义的CUDA、cuDNN的安装与卸载 ，

顺利地走到了安装成功的一步，接下来再去搜 torch 安装。

冇得支持cuda11.0的！！再根据 cuda11.0 +torch 的劝退文  找到 10.2版本的下载地址，

附cuda和cudnn链接：
cuDNN: https://developer.nvidia.com/rdp/cudnn-download
CUDA10.2 : https://developer.nvidia.com/cuda-10.2-download-archive

其它版本的 cuDnn : https://developer.nvidia.com/rdp/cudnn-archive

卸载cuda_V11

先打开【控制面板】卸载掉cuda11.0 （根据安装时间和名称含有cuda11的），
只保留 【NVIDIA的图形驱动程序、NVIDIA Physx系统软件，NVIDIA GeForce Experience】，如果你有这3个软件，就别卸载。

安装cuda_V10

下载CUDA10.2 之后重新自定义安装（链接同上）。

【注意：为了避免多次安装CUDA 后，安装目录竟然消失了，重新安装了三遍，还是消失？的问题，参考cuda安装教程+cudnn安装教程 】：临时解压目录千万不要和cuda的安装路径设置成一样的，否则安装结束，会找不到安装目录的！！

选择【自定义安装】 ：不要勾选“INVIDIA GeForce Experience co...”

 

 安装结束后， 查看环境变量，

 

添加   %CUDA_path%;%CUDA_PATH%\bin;%CUDA_PATH%\lib\x64;%CUDA_path%\libnvvp;  到path。

打开cmd验证安装成功

Win + R 打开 cmd 
cmd里查看CUDA 版本信息 nvcc -V
 cmd 查看GPU运行时的监测界面 nvidia-smi

运行几个相关的 exe
1. cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite
2. 运行bandwidthTest.exe
3. 运行deviceQuery.exe

安装cuDnn 

 cuDNN下载链接  https://developer.nvidia.com/rdp/cudnn-download

 

安装pytorch + torchvision

 

建议在上述命令后加上 清华镜像   -i https://pypi.tuna.tsinghua.edu.cn/simple 

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

网速差，可自行下载 torch_stable 的whl 文件 https://download.pytorch.org/whl/torch_stable.html
我的是用迅雷是 https://download.pytorch.org/whl/cu102/torch-1.6.0-cp36-cp36m-win_amd64.whl 

验证安装是否成功，打开pycharm 的python console 

import torch
print(torch.__version__)

import torchvision
print(torchvision.__version__)

OK,完啦。 

参考链接：
1. Win10中CUDA、cuDNN的安装与卸载  https://blog.csdn.net/XunCiy/article/details/89070315

2. 安装win10 + CUDA 11.0 + cuDNN8.0.2 + Pytorch-gpu(踩坑说明） https://my.oschina.net/u/4392911/blog/4480137 
3.  win10 卸载cuda https://blog.csdn.net/u013066730/article/details/85054164

4. cuda安装教程+cudnn安装教程 https://blog.csdn.net/sinat_23619409/article/details/84202651 

5.win10+pytorch+cuda安装 https://blog.csdn.net/lian740930980/article/details/107965740