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