ubuntu--搭建tensorflow-gpu环境

1. 安装python环境

sudo apt-get install python3-dev python3-pip

这里如果提示无法安装

需要添加新的源

sudo apt-get install software-properties-common

pysudo apt-add-repository universe

sudo apt-get update

2.安装virtualenv用于创建虚拟环境

sudo pip3 install -U virtualenv

3.创建一个environments文件夹，用于存放虚拟环境

mkdir environments

4.进入文件夹，创建一个虚拟环境tf2_py3

cd environments/

virtualenv --system-site-packages -p python3 ./tf2_py3

5.激活环境

source tf2_py3/bin/activate

6.安装tensorflow-gpu

https://www.tensorflow.org/install/source#common_installation_problems

查看cuda对应的tensorflow的版本：

这里我安装的cuda是10.0，所以对应安装tensorflow-gpu==2.0.0

pip install --upgrade tensorflow-gpu==2.0.0

7.安装成功，开始测试

python

import tensorflow as tf

print(tf.__version__)

tf.test.is_gpu_available()

显示true，表示安装成功。若失败，重启试试。

8.安装一些常用的包（可选）

pip install numpy matplotlib sklearn pandas jupyter

补充：cuda安装参考https://www.jianshu.com/p/762d6b134110