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

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

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

检查是否支持CUDA

在安装前,需要查看自己的电脑是否支持CUDA
检查方法:
1.桌面任务栏->显示隐藏的图表->英伟达控制面板图标(就是开始栏所在的导航栏中的上三角图标(大多在屏幕下端右侧))
或 打开控制面板->找到NVIDIA控制面板
在这里插入图片描述
2.在NVIDIA控制面板中->帮助->系统信息
在这里插入图片描述
3.选择组件,查看产品名称中含有CUDA的即为支持CUDA,可以继续下述工作,其后数字即为支持版本,兼容向下版本,例如,我的版本是11.1,那麽我可以下載<=11.1版本的CUDA。
在这里插入图片描述

安装CUDA

1.官网下载即可:https://developer.nvidia.com/cuda-toolkit-archive
选择需要下载的版本即可,我由于项目需求选择10.2版本在这里插入图片描述
在这里插入图片描述
2.安装:下载完成后,运行,弹出如下提示,该提示是暂时存放地址,不用修改,保持默认就好。在这里插入图片描述
3.完成后正式开始安装,选择自定义安装,俺在安装时忘记截图了,总之 自定义中,选择安装位置千万不要改变!!!,一路向下就好了,安装需要一点时间。
4.安装完成后配置环境变量(非所有人需要的工作),至此,开始出现分支情况,正常情况安装完成后会自动写入系统路径中,
此电脑右键->属性->高级系统设置->环境变量,检查,若在系统变量中已有路径,无需执行此步。(见下图内容)
在这里插入图片描述
在这里插入图片描述
若没有,在系统变量->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一定要大写)
,显示下图内容,为成功在这里插入图片描述

安装cuDNN

1.官网下载:https://developer.nvidia.com/rdp/cudnn-download
该网站需要注册/登录后下载,按照提示注册即可在这里插入图片描述
注册后下载需要版本即可在这里插入图片描述
2.解压并修改,将解压后的文件改名为cudnn,复制粘贴到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2,该地址是刚刚安装目录,地址最后版本对照自己所下版本
在这里插入图片描述

在这里插入图片描述
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版本
2.创建完成后conda activate pytorch_gpu,进入创建的环境,如图前面括号内的即为所在环境名称。在这里插入图片描述3.将官网下载命令输入第2步打开的环境命令行中:https://pytorch.org/get-started/locally/#no-cuda-1
该下载并不是可执行文件或者压缩包,是红线部分的命令,在第2步打开的环境命令行中输入安装
记得选择pip安装!!
在这里,我在阅读其他优秀博主文章时发现很多博主提到了速度慢的问题及解决方案,如果有需求大家可以自行搜索,因为我没有遇到不能给大家建议。
在这里插入图片描述
4.测试:在上面创建的pytorch_gpu环境命令行中 输入 python 进入python。
输入

import torch
torch.cuda.is_available()

意为 cuda是否可用,若显示TRUE,即为成功
在这里插入图片描述
至此,全部完成工作,希望大家在实战过程中注意细节!欢迎大家交流补充指正!本人较弱,请多多指教!
本文参考:https://blog.csdn.net/zazazaz1/article/details/108085918

  • 41
    点赞
  • 241
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 11
    评论
对于安装 CUDA 10.2PyTorch,你可以按照以下步骤进行操作: 1. 首先,确保你的显卡支持 CUDA 10.2,并且已经安装了相应的驱动程序。 2. 下载并安装 CUDA Toolkit 10.2,可以从 NVIDIA 官方网站下载对应版本的 CUDA Toolkit 安装程序。 3. 安装完成后,将 CUDA 的路径添到环境变量中。在 Windows 系统中,你可以右键点击“我的电脑”,选择“属性”,然后点击“高级系统设置”,在弹出窗口中点击“环境变量”,找到“系统变量”中的“Path”变量,点击“编辑”,在变量值的最后CUDA安装路径(例如:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin)。 4. 安装 Anaconda(如果未安装)。你可以从 Anaconda 官方网站下载适合你系统的安装程序,并按照指示进行安装。 5. 打开 Anaconda Prompt(或命令行界面),创建一个新的 conda 环境并激活它: ``` conda create -n myenv python=3.7 conda activate myenv ``` 6. 在激活的环境中,使用以下命令安装 PyTorch: ``` conda install pytorch torchvision cudatoolkit=10.2 -c pytorch ``` 这将会安装适用于 CUDA 10.2PyTorch 和 torchvision。 7. 安装完成后,你可以在 Python 中导入 PyTorch 来验证安装是否成功: ```python import torch print(torch.__version__) ``` 如果没有出现错误信息,并且能够正确显示 PyTorch 的版本号,则说明安装成功。 希望这些步骤对你有帮助!如有其他问题,请随时提问。

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

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Ang_go

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

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

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

打赏作者

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

抵扣说明:

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

余额充值