Windows10中安装GPU版本的Pytorch

最新推荐文章于 2024-07-05 19:45:13 发布

ahaha_zzzz

最新推荐文章于 2024-07-05 19:45:13 发布

阅读量1.7k

点赞数 1

文章标签： python windows cuda

本文链接：https://blog.csdn.net/star_xxzeng/article/details/108920264

版权

本文详细介绍了在Windows10系统中安装GPU版本Pytorch的步骤，包括检查CUDA版本、下载和安装CUDA及cuDNN，以及通过pip安装Pytorch。在安装过程中需要注意CUDA和cuDNN版本与Pytorch的兼容性，并正确配置环境变量。

摘要由CSDN通过智能技术生成

首先需要说明的是，这个教程是在已经安装好Anacoda和PyCharm的前提下进行的，并且需要一个已经配置了基础库的虚拟环境。我是在cmd中通过conda的命令创建了名为pytorch_gpu01的虚拟环境，其中的Python版本是3.6 。在此基础上，依照下面的步骤进行安装Pytorch。

以下教程中如果有不妥当之处，欢迎各路大侠指出！

1. CUDA和cuDNN的下载和安装

下载CUDA和cuDNN之前，先检查一下电脑所支持的CUDA版本。如果电脑所支持的CUDA版本在Pytorch中没有对应的，这一步可以直接跳过，不安装CUDA和cuDNN。

注：安装了CUDA和cuDNN的是为了安装、配置GPU版本的Pytorch，否则可以直接安装CPU版本的Pytorch。

电脑桌面空白处，右键，点击NVIDIA控制面板，选项卡中选择“帮助”——“系统信息”——“组件”即可看到cuda的版本。
如下图所示，我使用的电脑所支持的版本是CUDA10.2
查看电脑所支持的CUDA版本

打开Pytorch官网，可以看见，Pytorch中是有对应的CUDA版本的。因此，可以安装GPU版本的Pytorch。

在这里插入图片描述

如果电脑所支持的CUDA版本低于Pytorch中可支持的版本，安装了GPU版本的也使用不了。但如果电脑可支持的CUDA版本高于Pytorch可支持的版本，也可以安装GPU版本的Pytorch。具体情况如何，就不在本文的范围内了。

因此，接下来就是下载和安装CUDA和cuDNN。

CUDA下载地址：nvidia官网
cuDNN下载地址：nvidia官网
其中，cuDNN下载时需要注册，登录，之后填写一个简单的问卷，即可进行下载对应的cuDNN。
如下图，选择对应版本的CUDA和cuDNN进行下载。

关注