1、新建环境，避免与基础环境出现冲突

conda create -n pytorch_gpu python=3.7

2、切换到pytorch环境

conda activate pytorch_gpu

3、安装基本框架

conda install pandas
conda install jupyter notebook
......

4、安装pytorch

1）进入官网https://pytorch.org/get-started/locally/

2）选择相应的系统 和cuda版本（查看cuda版本，见4）

3）复制上图中最后一行代码到pytorch环境终端，（注意：-c pytorch删掉，不然会从官网下载安装，很慢）

# 此处一定要注意：python和cuda在清华源中与是有版本对应关系的，不能随意安装版本：如果源中是python3.7 + cuda10.1，那么如果你创建的环境是python3.6 + cuda10.1，安装会很慢，不会从清华源下载
conda install pytorch torchvision cudatoolkit=10.1

 查看清华源中版本对应关系的网址：https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

cudnn安装包地址：https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

4）cuda版本查看：

a）windows：

桌面右键-->NVDIA控制面板-->帮助-->系统信息

 

组件-->NVCUDA.DLL查看CUDA版本

 

 

b）linux：

打开终端，输入：nvcc -V

或者：

CUDA:

cat /usr/local/cuda/version.txt

cudnn:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2