Anaconda pytorch GPU版本安装实践

创建python环境

我使用的是Anaconda安装，也建议使用Anaconda，因为使用起来比较方便。若是没有安装的话可以直接进Anaconda官网直接下载安装。

首先要使用Anaconda创建一个虚拟环境，如下图所示，然后选择python3.7.1，我将这个环境命名为pytorch

安装pytorch

下面就进入安装正题了，先进pytorch官网选择合适的版本，如下图所示：

如上图，以往我使用的是CUDA=10.2的版本安装的，但是官网里面不提供这个版本了，让提示安装CUDA=11.6的，（这个根据自己电脑的实际情况选择）因此使用点击CUDA=11.6，复制命令：conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

然后打开anaconda prompt，切换到刚刚创建的虚拟环境中：

然后输入刚刚复制的命令，因为我是挂的VPN下载的，所以速度感觉还可以，若是嫌速度慢的话可以把-c pytorch删掉，换用其他的源：

接下来输入y，然后等待即可，但下载途中我这里出了点问题：

这个库可能因为网络原因导致没下载成功，不担心，下载结束后再次运行一次刚刚复制的命令conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge即可。下载成功后输入python进入python环境进行验证，如下所示：

当结果是true的时候就表示安装成功了。