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
欢迎关注公众号，一起交流python技术。
