Windows10下的Pytorch安装教程
首先需要说明的是,这个教程是在已经安装好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。
查看电脑CUDA版本
电脑桌面空白处,右键,点击NVIDIA控制面板,选项卡中选择“帮助”——“系统信息”——“组件”即可看到cuda的版本。
如下图所示,我使用的电脑所支持的版本是CUDA10.2
查看Pytorch中可支持的CUDA版本
打开Pytorch官网,可以看见,Pytorch中是有对应的CUDA版本的。因此,可以安装GPU版本的Pytorch。
如果电脑所支持的CUDA版本低于Pytorch中可支持的版本,安装了GPU版本的也使用不了。但如果电脑可支持的CUDA版本高于Pytorch可支持的版本,也可以安装GPU版本的Pytorch。具体情况如何,就不在本文的范围内了。
因此,接下来就是下载和安装CUDA和cuDNN。
1.1 下载
CUDA下载地址:nvidia官网
cuDNN下载地址:nvidia官网
其中,cuDNN下载时需要注册,登录,之后填写一个简单的问卷,即可进行下载对应的cuDNN。
如下图,选择对应版本的CUDA和cuDNN进行下载。