小白win11+cuda12.1安装pytorch避坑指南

Tips:如果C盘足够大(大于200G),可以考虑装C盘,否则建议装其他盘。安装完anaconda和pytorch,大概需要23G,还没有算后续的训练和数据。

先说一下我个人配置,win11操作系统,x64(现在很少32位的系统了),RTX4060显卡,安装anaconda+pytorch步骤如下:

1、安装cuda和cuDNN

首先系统支持的cuda,打开nvidia控制面板,点击左下方的“系统信息”,点击“组件”,显示版本是“12.1.112”。

到cuda官网下载相应版本,一般是向下兼容(低一点的版本是可以的)

kCUDA Toolkit Archive | NVIDIA Developer

网上很多下载及安装教程,这里不累赘。但是有个tips:第一次安装,以下页面建议选择“自定义”选项(借用大神的图)

自定义选项里不要勾选“Visual Studio”(前面步骤没有装的话,这又是一个tips:有人建议先到微软官网安装Visual Studio,但是后来发现不安装也能用)

安装完成后,大部分需要手动添加环境变量,方便检测是否安装成功。

win11是在桌面右键,点击“显示设置”,点击左边选择“系统”选项,选择右边的“系统信息”选项,“高级系统设置”-“高级”-“环境变量”-“系统变量”-”path”-“编辑”-“新建”,把相关内容加进去。(一共四个环境变量,一般是自动生成2个cuda_path和cuda_path_v12.1,另外两个手动添加)

至此,cuda安装完成,启动cmd,输入nvcc –version可以查看版本号,输入set cuda可以查看cuda设置的环境变量。

安装cuDNN。cuDNN不是一个安装软件,就是三个最新的文件夹,替代cuda里面的文件,下载,复制,粘贴到cuda安装目录即可。不装cuDNN也可,只是性能没那么好。

2、安装anaconda

推荐官网下载和安装,最新版本是anaconda3+python3.11,个人选择。

Tips:小白一定要全部勾选,特别第二项“add anaconda3 to my path environment variable”,如果没有这一项,小白会在后面的步骤痛不欲生。大神们可以忽略。

3、安装pytorch

打开anaconda prompt,按照网上教程一步步:
1)创建环境:conda create -n pytorch_gpu python=3.11
   环境名加gpu后缀是为了提醒自己计算机有GPU
2)激活环境:conda activate pytorch_gpu

3)强烈建议使用官网的pip方法安装,不要选择conda选项(掉了好几次坑),也不要用清华镜像,最好在夜深人静时候安装,网速比较快(这个大概2.3G),我白天下载用了2个小时,比较痛苦。

安装完成,就可以打开prompt,做以下操作:

  1. )输入python
  2. )输入import torch,没有报错说明安装成功。
  3. )输入print(torch.__version__),显示版本信息
  4. )输入print(torch.cuda.is_available()),返回如果是True,恭喜你安装完成。

4、在prompt里面显示已经安装pytorch(或者torch),但是spyder里面import torch报错,这个时候真的怀疑人生了。通过学习大神的经验,做以下操作:

1)、打开anaconda NAVIGATOR,选择Environmeng—base(root)—右边三角形—Open Terminal,会打开系统的cmd,重新做第三步的pip安装,这次会很快,因为已经系统已经检测到下载完成。

2)、安装完成,打开spyder,做第三步同样的测试。

测试完成,同时在spyder里面能用。安装完成,耗时12小时(白天)。

  • 26
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值