一文多图搞定Ubuntu安装Anaconda+GPU Driver+CUDA+CUDNN+TensorFlow-gpu+Pytorch(附网盘链接)

1 Anaconda

1.1 下载 Anaconda

通过国内镜像源下载，请点击清华开源软件镜像站 ，这里我选择 Ananconda3, python 版本为 3.6，即 Anaconda3-5.0.1-Linux-x86_64.sh，如下图所示。

1.2 安装 Anaconda

先切换到 anaconda 安装包所在的路径下，接着执行以下命令

chiyukunpeng@chiyukunpeng:~$ cd anaconda3
chiyukunpeng@chiyukunpeng:~/anaconda3$ sh Anaconda3-5.0.1-Linux-x86_64.sh   

接下来一路按回车以及输入 yes 即可。重启终端，输入以下命令验证安装是否成功

chiyukunpeng@chiyukunpeng:~$ source ~/.bashrc
chiyukunpeng@chiyukunpeng:~$ python  # 出现以下 Anaconda 的 python 环境即可
Python 3.6.3 |Anaconda, Inc.| (default, Oct 13 2017, 12:02:49) 
[GCC 7.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> exit()  # 输入退出此环境

1.3 conda换源

Anaconda 是深度学习环境搭建的必备工具，但服务器在国外，下载速度很慢，换为国内源十分重要。
输入以下命令，打开 condarc 文件

chiyukunpeng@chiyukunpeng:~$ sudo gedit ~/.condarc

然后将以下内容复制粘贴更换掉此文件内所有内容，保存退出即可。

auto_activate_base: false
channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
show_channel_urls: true

2 GPU 驱动

2.1 查看 GPU

在终端输入以下命令，查看 GPU 型号及推荐驱动

chiyukunpeng@chiyukunpeng:~$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001C20sv000017AAsd000039F5bc03sc00i00
vendor   : NVIDIA Corporation
model    : GP106M [GeForce GTX 1060 Mobile]  # 我的GPU型号
driver   : nvidia-driver-390 - third-party free
driver   : nvidia-driver-435 - distro non-free
driver   : nvidia-driver-410 - third-party free
driver   : nvidia-driver-415 - third-party free
driver   : nvidia-driver-440 - third-party free recommended  # 推荐的驱动型号
driver   : xserver-xorg-video-nouveau - distro free builtin

2.2 安装驱动

在终端输入以下命令

chiyukunpeng@chiyukunpeng:~$ sudo apt-get install nvidia-driver-440

安装成功后，打开桌面左下角的应用程序抽屉，打开软件和更新->附加驱动，选择刚刚下载的驱动，如下图所示
重启电脑，打开终端，输入以下命令验证是否安装成功

chiyukunpeng@chiyukunpeng:~$ nvidia-smi
Sun Apr 19 21:39:38 2020       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.82       Driver Version: 440.82       CUDA Version: 10.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 1060    Off  | 00000000:01:00.0 Off |                  N/A |
| N/A   45C    P5    10W /  N/A |    245MiB /  6078MiB |      3%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |