CUDA(10.2)+PyTorch安装加配置 详细完整教程

以下均为博主亲测的可靠流程,其中有一些我遇到的问题及解决方法,希望能帮到大家,不再踩坑,也欢迎大家交流遇到的问题和配置方法。

CUDA(10.2)+PyTorch安装加配置 详细完整教程

检查是否支持CUDA

在安装前,需要查看自己的电脑是否支持CUDA
检查方法:
1.桌面任务栏->显示隐藏的图表->英伟达控制面板图标(就是开始栏所在的导航栏中的上三角图标(大多在屏幕下端右侧))
或 打开控制面板->找到NVIDIA控制面板

CUDA(10.2)+PyTorch安装加配置 详细完整教程
2.在NVIDIA控制面板中->帮助->系统信息
CUDA(10.2)+PyTorch安装加配置 详细完整教程
3.选择组件,查看产品名称中含有CUDA的即为支持CUDA,可以继续下述工作,其后数字即为支持版本,兼容向下版本,例如,我的版本是11.1,那麽我可以下載

; 安装CUDA

1.官网下载即可:https://developer.nvidia.com/cuda-toolkit-archive

CUDA(10.2)+PyTorch安装加配置 详细完整教程
CUDA(10.2)+PyTorch安装加配置 详细完整教程
CUDA(10.2)+PyTorch安装加配置 详细完整教程
3.完成后正式开始安装,选择自定义安装,俺在安装时忘记截图了,总之 自定义中, 选择安装位置千万不要改变!!!,一路向下就好了,安装需要一点时间。
4.安装完成后配置环境变量(非所有人需要的工作),至此,开始出现分支情况,正常情况安装完成后会自动写入系统路径中,
此电脑右键->属性->高级系统设置->环境变量,检查,若在系统变量中已有路径,无需执行此步。(见下图内容)
CUDA(10.2)+PyTorch安装加配置 详细完整教程
CUDA(10.2)+PyTorch安装加配置 详细完整教程
若没有,在系统变量->path中加入以下内容
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\CUPTI\lib64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2\bin\win64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2\common\lib\x64

5.检查安装是否成功,cmd中输入 nvcc -V(V一定要大写)

CUDA(10.2)+PyTorch安装加配置 详细完整教程

安装cuDNN

1.官网下载:https://developer.nvidia.com/rdp/cudnn-download

CUDA(10.2)+PyTorch安装加配置 详细完整教程
CUDA(10.2)+PyTorch安装加配置 详细完整教程
2.解压并修改,将解压后的文件改名为cudnn,复制粘贴到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2,该地址是刚刚安装目录,地址最后版本对照自己所下版本
CUDA(10.2)+PyTorch安装加配置 详细完整教程

CUDA(10.2)+PyTorch安装加配置 详细完整教程
3.可选步骤,博主的安装并不需要此步骤即可正确后续,有朋友反应需要配置,自行选择吧
在系统变量->path中加入以下内容
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\CUPTI\lib64

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\cudnn\bin

安装PyTorch

1.打开anaconda prompt(有关anaconda的安装和使用,大家可以去我主页找,这里不再赘述)
为pytorch创建一个虚拟环境,与base等区分开

conda create -n pytorch_gpu python=3.9

pytorch_gpu是环境名称,可自行选取,python=3.9是选择的python版本,可自行选择,conda会自动下载选择的python版本

CUDA(10.2)+PyTorch安装加配置 详细完整教程
该下载并不是可执行文件或者压缩包,是红线部分的命令,在 第2步打开的环境命令行中输入安装
记得选择pip安装!!
在这里,我在阅读其他优秀博主文章时发现很多博主提到了速度慢的问题及解决方案,如果有需求大家可以自行搜索,因为我没有遇到不能给大家建议。
CUDA(10.2)+PyTorch安装加配置 详细完整教程
4.测试:在上面创建的pytorch_gpu环境命令行中 输入 python 进入python。
输入
import torch
torch.cuda.is_available()

意为 cuda是否可用,若显示TRUE,即为成功

CUDA(10.2)+PyTorch安装加配置 详细完整教程
至此,全部完成工作,希望大家在实战过程中注意细节!欢迎大家交流补充指正!本人较弱,请多多指教!
本文参考:https://blog.csdn.net/zazazaz1/article/details/108085918

Original: https://blog.csdn.net/Ang_go/article/details/122329715
Author: Ang_go
Title: CUDA(10.2)+PyTorch安装加配置 详细完整教程

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/716770/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球