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

目录

安装 cuda11 (坑)

卸载cuda_V11

安装cuda_V10

安装cuDnn 


安装 cuda11 (坑)

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

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

win10 安装CUDA 11.0后再安装GPU版torch的踩坑记录_第1张图片

然后根据网上的教程(为避免报错,不安装 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 之后重新自定义安装(链接同上)。

win10 安装CUDA 11.0后再安装GPU版torch的踩坑记录_第2张图片

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

win10 安装CUDA 11.0后再安装GPU版torch的踩坑记录_第3张图片

 安装结束后, 查看环境变量,
win10 安装CUDA 11.0后再安装GPU版torch的踩坑记录_第4张图片

 

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

win10 安装CUDA 11.0后再安装GPU版torch的踩坑记录_第5张图片

打开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

win10 安装CUDA 11.0后再安装GPU版torch的踩坑记录_第6张图片

 

建议在上述命令后加上 清华镜像   -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__)

win10 安装CUDA 11.0后再安装GPU版torch的踩坑记录_第7张图片

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 

你可能感兴趣的:(其它技巧)