在anaconda中安装pytorch的GPU版本

本文前提：

1.你已经下载好了anaconda,最好是新建一个虚拟环境来安装pytorch的GPU版本，并且设置了国内镜像源；
2.了解自己电脑对应的cuda版本，可通过nvidia-smi，并下载好了cuda。

安装pytorch的GPU版本

到官网中https://pytorch.org/首页向下翻，找到以下内容，然后根据自己的cuda版本复制相应的命令。安装包相对较大，请耐心等待。

安装完成后，通过conda list可查看已下载的依赖：

gpu版本能够看到cuda相应的依赖包，而以下是非gpu版本：

你也可以通过查看以下的输出查看是否安装成功：

torch.__version__
torch.cuda.is_available()
torch.cuda.get_device_name(0)