Ubuntu 20.10通过 Anaconda安装Tensorflow-GPU版本

简介

安装tensorflow-gpu，需特别注意tensorflow-gpu、Python、CUDA、cuDNN版本的适配信息，版本不适配会导致tensorflow-gpu安装失败。版本适配信息可参考官网：链接: https://tensorflow.google.cn/install/source_windows.

通过Anaconda安装tensorflow较为方便，不用单独安装CUDA、cuDNN，避免出现未知错误

安装步骤

一. Nvidia显卡驱动安装

这里我就默认你的Ubuntu系统已经安装好啦。我们先要安装显卡驱动。最方便的办法就是直接打开系统中软件与更新，再点击附加驱动，选择一个最新的Nvidia专用即可。

二. Anaconda 安装

1.下载

官方下载地址：
链接: https://www.anaconda.com/products/individual.
页面最下方找到Linux版本，选择第一个下载。

2、安装

找到下载好的压缩包所在文件夹，打开终端，执行命令：

bash Anaconda3-2020.02-Linux-x86_64.sh

一路回车+yes即可

3、验证

在终端输入下面命令查看python版本：

python3

若返回python版本信息说明安装成功。

4、配置环境变量

在终端输入命令：

gedit ~/.bashrc

在终端输入命令：

gedit ~/.bashrc

打开文件后，在最后一行添加语句：

export PATH=“/*自己的安装路径*/anaconda3/bin:$PATH”

点击保存，关闭bashrc文件，最后一定要在终端输入下面命令生效

source ~/.bashrc

5.验证

打开终端，输入conda -V
若返回conda版本信息，说明安装成功。

三. Tensorflow-GPU版本安装

1.新建python虚拟环境

截至我写这篇博客的时间，最新版本的Anaconda默认是python3.8版本，但是Tensorflow还没有支持到3.8

我是新建了一个python3.6的环境，具体步骤如下。
在此建议将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/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

# 设置搜索时显示通道地址
conda config --set show_channel_urls yes

在终端输入：

conda create -n python36 python=3.6 anaconda

新建一个名为python36的虚拟环境，python版本为3.6
输入conda activate python36激活环境。
输入python --version查看此时的python版本

2. 安装Tensorflow-GPU

查找满足要求的tensorflow版本

conda search tensorflow-gpu

系统会返回可供下载的版本，我选择的是2.2.0版本。

输入conda install tensorflow-gpu==2.2.0，等待安装完毕
到此，我们就安装好了，可以输入以下代码进行测试

python
import tensorflow as tf
print(tf.__version__)