Windows10 配置PyTorch环境

安装CUDA

确定CUDA版本号

打开控制面板：
再去官网比对自己可以用什么版本的CUDA Toolkit。我自己是496.13版本的驱动，所以CUDA的这些版本我都可以使用
查看Pytorch支持的CUDA版本

现在比较新的Pytorch版本支持的是CUDA11.3和CUDA10.2
若是自己的驱动不支持这两个的可以点击Previous version of PyTorch的按钮去查看之前的版本，以及之前的版本支持的CUDA版本

下载CUDA

确定好了版本号后就开始下载，下载地址。在选择Installer Type的时候，根据自己的需要去选择下载Local本版本还是Network版本。一般选择Local版本即可
后续安装一路下一步即可，需要自定义安装的就选自定义安装

检验CUDA是否安装成功

测试CUDA是否安装成功，在PowerShell或者CMD中输入nvcc -V，输出结果如下即为安装成功：

安装cuDNN

下载

直接去官网，找对应的版本下载就好了。（PS：因为没有直接标明对应CUDA11.3的，下载CUDA 11.X的就好了）

操作

下载完后是一个压缩包，解压后是一个名为cuda的文件夹，将cuda中的bin文件夹、include文件夹以及lib文件夹中的内容复制到安装CUDA的目录下的同名文件夹中。
（PS：注意不是直接将三个文件夹复制到CUDA目录中，而是将各自内容复制到对应文件夹中,一般安装路径为：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA%版本号%)

cuda\bin\* ====> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA%版本号%\bin\
cuda\include\* ====> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA%版本号%\include\
cuda\lib\x64\* ====> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA%版本号%\lib\x64\

验证cuDNN是否安装成功

进入到cuda的安装路径，C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\extras\demo_suite，找到如下两个.exe文件：

在Powershell或者CMD中运行，若是安装成功，运行结果如下：

安装PyTorch

直接使用pip或者conda命令安装。conda和pip需要换源，安装和换源教程大家就自己去找了

进入PyTorch官网，选择自己的条件和配置，并使用PyTorch中给出的命令，直接在Powershell或者CMD中运行即可。若是配置没有在首页显示，可以去历史版本中寻找命令。（切记：命令行安装时，需要把-c参数删除，否则就是从官方源下载，国内安装速度感人）。

安装后可能会提示需要删除几个文件，按照要求删除即可。(不知道删了会不会影响，我做了备份，目前无影响)。

验证PyTorch是否能使用GPU

# 运行该代码
import torch

print(torch.__version__)
print(torch.cuda.is_available())

若是运行结果如下：

则成功。

大家也可以跑一个小例子，看看效果。demo教程

参考文章：
第一个pytorch程序（Pycharm）
【CUDA】cuda安装（windows版）
Windows10安装Pytorch的完整安装教程
 Windows10检查Cuda和cuDNN是否安装成功？

__EOF__

本文作者：Fortunater
本文链接：https://www.cnblogs.com/Fortunater/p/15768912.html
关于博主：不急不躁，一步一步走向技术巅峰
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！