admin管理员组文章数量:1646237
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
目录
前言
一、CUDA安装
二、cuDNN安装
三、配置环境变量
前言
有一天跑yolo系列,出现cuda无效的bug,然后更新了环境,发现训练速度很慢,后来发现是自己搞错了环境,弄成CPU版本的了。所以就综合一下网上的教程和自己踩得坑,写一篇pytorch(GPU版本)详细教程,也是希望自己之后弄崩环境后能有个参考吧!
一、CUDA安装
1.CUDA驱动查看
电脑桌面右键,打开NVIDIA控制面板
点击左下角系统信息
点击组件,查看nvidia cuda驱动版本
可以看到我的是11.6,目前来说11.6已经够用了,但是如果你的版本相对较低,建议更新一下驱动版本。
2.CUDA驱动更新
首先进入nvidia官方驱动的官网:官方驱动 | NVIDIA
然后点击搜索
看一下这里的型号是不是和自己电脑的GPU一样的,不一样就换成和自己GPU一样的型号即可。然后点击搜索后出现下面的下载界面,点击下载。
将下载好的驱动点击安装,安装步骤就是默认下一步(安装地址可以根据自己电脑实际情况来,默认c盘)。
更新好驱动后,继续查看驱动版本。同时按住win+R键,输入cmd进入终端,输入nvidia-smi,查看对应版本。 可以看到此时我的驱动版本对应的是512.15,cuda版本对应的11.6。
3.CUDA安装
(1)查看安装版本
首先进入CUDA Toolkit 官方界面:CUDA 12.1 Release Notes
向下划找到以下表格。
我的驱动版本是512.15,除了前两个cuda12以上的不能安装,其他都可,但是前面cuda版本是11.6,所以要安装的cuda版本不能超过11.6!这里作者选择cuda11.6版本进行安装。(下载的CUDA版本一定要是在11.6以下。)
(2)CUDA下载
CUDA官网下载地址:CUDA Toolkit 12.1 Downloads | NVIDIA Developer
进来之后点击查看历史版本
进来之后找到自己想安装的CUDA版本,本文选的是cuda11.6.0
点击相应的版本后,win10就选10,win11系统就选11,选择相应版本如下图即可下载
(3)CUDA安装
点击下载好的.exe文件,系统检查点击下一步,接下来安装选项,选择自定义,然后点击下一步。
选择安装地址的时候,可以选择默认的,也可以安装在其他地址,但一定要记住这个地址,因为后面配置环境的时候要用到这个地址!!!
后面就进行下一步直到安装结束就可以了。
查看是否安装成功,同时按住win+R键,输入cmd,进入终端,输入nvcc -V,如果输出如下,即安装成功。
二、cuDNN安装
进入到cuDNN官网:https://developer.nvidia/rdp/cudnn-download
点击Archive of previous Releases。如果上面链接点击后没出现这个界面,就要注册一个账号,然后再登录就到这个界面啦。
接着点击cuDNN8.x-1.x。
查找CUDA11.6对应cuDNN版本,只要是11.x的都可以,我这里选的是8.5.0。其他版本也可以
下载,解压,之后将cuDNN文件中这三个文件夹中的内容分别复制到cuda的对应文件夹中
cuda的安装地址在前面安装的时候是有的,可以截图。或者默认。就是前面这个地方的安装地址
三、配置环境变量
首先在设置里面搜索环境变量,打开编辑系统环境变量。
打开环境变量。
找到系统变量,path,点击
将下面的cuda安装路径新建到环境变量中,注意,这里是根据自己的安装路径填。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib\x64
添加后,点击确定。即可完成。
版权声明:本文标题:新手保姆级GPU版本pytorch安装(win系统),CUDA和torch对应版本! 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:https://m.elefans.com/xitong/1729420128a1200518.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论