当前位置: 代码网 > 科技>操作系统>Windows > CUDA配置及Pytorch-gpu安装教程

CUDA配置及Pytorch-gpu安装教程

2024年07月28日 Windows 我要评论
c)点击对应版本后,会进入下一个页面,操作系统(Operating System)选择Windows,构架(Architecture)选择x86_64,版本(Version)选择11,安装类型(Installer Type)选择exe(local),然后点击下载按钮。d)下载完成后双击安装,一路同意或者确认即可,选择自定义安装(第一个选择路径无所谓在哪,只是提供解压路径,安装完成后会自动删除,第二个路径是安装的位置,可以自己修改,但是需要记住安装位置。)首页,往下滑,选择对应选项,复制方框内的命令。

1.查看电脑显卡驱动版本

a)按win键和r键,调出系统的运行窗口

b)输入cmd,回车

c)进入命令行窗口后,输入:nvidia-smi,查看版本信息

如图所示,本机的driver version: 555.85,cuda version: 12.5

2.下载cuda toolkit和cudnn

a)查看驱动版本对应表格(https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html),由1.c知道本机driver version: 555.85>=555.85

b)cuda toolkit:在官网(https://developer.nvidia.com/cuda-toolkit-archive)选择适合自己电脑的版本下载(由1.c可知本机cuda version: 12.5(具体是12.5.51,这个可以在nvidia控制面板-帮助-系统信息-组件处查看),版本>=cuda toolkit 12.5.0,所以我下载的最新版本cuda toolkit 12.5.0)。

c)点击对应版本后,会进入下一个页面,操作系统(operating system)选择windows,构架(architecture)选择x86_64,版本(version)选择11,安装类型(installer type)选择exe(local),然后点击下载按钮。

d)下载完成后双击安装,一路同意或者确认即可,选择自定义安装(第一个选择路径无所谓在哪,只是提供解压路径,安装完成后会自动删除,第二个路径是安装的位置,可以自己修改,但是需要记住安装位置。

e)cudnn:在官网(https://developer.nvidia.com/rdp/cudnn-archive下载对应的版本(因为上面安装的为cuda12.5,所以选择对应的cuda12.x),点开后选择第一个下载(注意,这个下载需要登陆nvidia账号,自己注册一下)。

 

f)下载完成后解压,按住ctrl键+c键,复制文件夹内的如下三个文件夹

g)找到上面安装位置(默认路径应该是c:\program files\nvldla gpu computing toolkit\cuda\v12.5,这里我设置的路径为d:\nvidia\cuda\v12.5)对应的文件夹,按ctrl键+v键粘贴,确认替换。

h)回到之前的命令行窗口,输入:nvcc -v(注意中间有一个空格),出现如下信息,表示安装成功。

3.创建虚拟环境

a)根据“note: latest pytorch requires python 3.8 or later.”,创建一个python3.8以后的环境

b)打开anaconda prompt,输入:conda create -n pytorch python==3.11(这里我设置虚拟环境名称为pytorch,指定python版本为3.11)

c)中间过程如下,输入:y并点击回车,开始创建虚拟环境

d)创建成功

4.安装pytorch-gpu

a)在pytorch官网(https://pytorch.org/)首页,往下滑,选择对应选项,复制方框内的命令

b)打开anaconda prompt,输入:conda activate pytorch,以激活pytorch虚拟环境,此时可以看到前面括号内的base变成了pytorch。

c)粘贴刚才复制好的命令,慢慢等待下载安装即可。

5.检查

a)打开anaconda prompt,输入:conda activate pytorch,进入虚拟环境

b)输入:python,进入python代码模式

c)导入torch,输入:import torch

d)输入:torch.cuda.is_available()

e)结果显示true说明安装成功。

(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com