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 

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
©️2022 CSDN 皮肤主题:技术工厂 设计师:CSDN官方博客 返回首页
评论 21

打赏作者

wamg潇潇

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值