anaconda3安装pytorch-GPU版

一. 查看电脑显卡的版本步骤为：此电脑右击–>管理–>设备管理器–>显示适配器。就可以看到电脑显卡的版本了。



二.安装pytorch
打开anaconda终端（按win键）

查看有哪些环境（新安装的anaconda只有一个base环境）

conda env list

创建虚拟环境conda create -n 环境名字（英文） python=x.x（python版本），如下，我就是创建了一个名字叫pytorch，python是3.7.1版本的环境。
Python版本可以自己查看是那个版本的

conda create -n PyTorch python=3.7.1

就会创建一个新的虚拟环境（环境名可以自己命名），这个虚拟环境会安装一些基础的包。询问是否安装的时候，输入y。就可以创建环境了。
执行如下命令，激活这个环境

activate PyTorch

安装pytorch-gup版的环境，由于pytorch的官网在国外，下载相关的环境包是比较慢的，所以我们给环境换源。在pytorch环境下执行如下的命名给环境换清华源。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

根据自己的cuda查找对应的pytorch版本
查看自己的cuda版本
cmd输入

nvcc --version

或者

安装命令（官方显示的有的可能没有对应的版本可以根据下面的网址找对应的cuda版本）

https://pytorch.org/get-started/previous-versions/

安装完后验证是否安装成功

import torch
print(torch.cuda.is_available())