一、安装 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 自己需要的包