Win11首先安装Vs2017,然后安装CUDA和cuDNN,最后安装pytorch用于深度学习训练--最全教程

 一、安装 Visual Studio (VS) 2017版本。

说在前面:无法从Visual Studio官网免费下载之前发布的版本。虽然官网说可以装VS之前的版本,但是需要订阅 Visual Studio Dev EssentialsDev Essentials

安装正常流程是进入VS的官网首页,划到最下面点击较早的下载,可以看到

但是点击下载,实际上跟吴无法下载Visual Studio Community 2017,你需要订阅。

只能通过从前保存的安装包下载,附上百度云下载链接,如果失效了请留言。

链接:https://pan.baidu.com/s/1P-v0j9XZvEInIbIUNvrgAg 
提取码:327h

注: 如何安装中遇到共享组件、工具和SDK无法更改路径的问题,可以安装

1. Win+R
2. 输入cmd
3. 输入regedit进入注册表,找到路径:HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\VisualStudio\Setup
4. 更改步骤3路径下的SharedInstallationPath项。

二、下载CUDA

截至到目前,pytorch官网发布的Windows 稳定版只到CUDA 12.1,

所以去CUDA官网Developer Download Centers | NVIDIA Developer)安装CUDA12.1, 通过以下步骤找到之前的版本。

三、下载cuDNN

点击官网地址CUDA Deep Neural Network (cuDNN) | NVIDIA Developer),点击Download cuDNN Library,找到与CUDA对应的版本下载。

四、安装CUDA和cuDNN

打开CUDA的exe文件进行安装,记住自己的安装路径,默认路径为

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\

cuDNN解压,放入CUDA的安装路径。

五、配置Pytorch

1. 打开 anaconda prompt

2. conda create –n pytorchGPU python=3.10

3. activate pytorchGPU

4. 

# CUDA 12.1
conda install pytorch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 pytorch-cuda=12.1 -c pytorch -c nvidia

5. pip 自己需要的包

  • 14
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值