Ubuntu16 安装GPU版TensorFlow

Ubuntu16 安装GPU版TensorFlow

安装GPU驱动

ubuntu-drivers devices                  # 查看支持的nvidia驱动版本
sudo apt-get purge nvidia*              # 删除当前的nvidia驱动
sudo apt-get install nvidia-430         # 安装固定版本的nvidia驱动

安装Anaconda

从Anaconda官网下载安装即可。

conda create --name TF python=3.6       # 创建Python版本为3.6的运行环境，命名为TF
conda activate TF                       # 激活TF运行环境
conda deactivate                        # 退出当前运行环境，进入root环境
conda env list                          # 显示当前存在的运行环境

conda list                              # 显示当前环境已安装的包
conda env export > TF.yml               # 将当前环境配置导出，生成TF.yml文件
conda env create -f TF.yml              # 从TF.yml文件创建环境
conda remove --name TF --all            # 删除TF环境和该环境下所有的包

conda install scikit-learn              # 安装scikit-learn包
conda remove --name TF package_name     # 删除TF环境下的某个包

安装GPU版TensorFlow

查询TensorFlow按章教程，选择要安装的版本。

conda install tensorflow-gpu=2.2.0     # 安装版本为2.2.0的tensorflow-gpu

测试

运行如下程序测试是否安装成功。

import tensorflow as tf
tf.test.is_gpu_available()