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

最后编辑于 :
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容