ubuntu20.04安装Pytorch,CUDA——日志（3）

把要安装的工具调查清楚后，就开始安装了。

首先用anaconda创建虚拟环境

conda create -n RL python=3.6

创建成功，激活虚拟环境。可以看到括号里的base已经变成了RL

(base) johnwatson@rescuer-r720:~$ source activate RL
(RL) johnwatson@rescuer-r720:~$ 

查询python版本号，可以看到Pyhton已经变成了3.6

(RL) johnwatson@rescuer-r720:~$ python -V
Python 3.6.13 :: Anaconda, Inc.

添加conda清华镜像

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

安装CUDA9.0

conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

安装成功，然后安装cuDNN7.6.5

conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

安装成功，然后安装Pytorch1.1.0，torchvision0.3.0

conda install pytorch=1.1.0 torchvision=0.3.0 -c soumith

安装成功。检查是否安装成功，输入以下代码

(RL) johnwatson@rescuer-r720:~$ python
Python 3.6.13 |Anaconda, Inc.| (default, Jun  4 2021, 14:25:59) 
[GCC 7.5.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True

返回True，表明安装成功。