赞
踩
目录
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.0 , 随便搜索下 cuda download,就会给您跳转到这个页面:
然后根据网上的教程(为避免报错,不安装 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
先打开【控制面板】卸载掉cuda11.0 (根据安装时间和名称含有cuda11的),
只保留 【NVIDIA的图形驱动程序、NVIDIA Physx系统软件,NVIDIA GeForce Experience】,如果你有这3个软件,就别卸载。
下载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下载链接 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
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。