python3搭建tensorflow的gpu版虚拟环境

背景:

本机以安装完成英伟达1080的驱动,更新到410以上版本,同时安装上gpu加速工具CUDA10。本机操作系统为ubuntu18,装有python3(3.8.6)。

第一步:安装虚拟环境库

打开终端,输入:
pip install virtualenv

屏幕截图.png

第二步:创建虚拟环境

创建python3虚拟环境,终端输入:
python3 -m venv tutorial-env

屏幕截图.png

激活虚拟环境:
source tutorial-env/bin/activate

屏幕截图.png

该行的前缀变化了!

退出虚拟环境:
deactivate

屏幕截图.png

前缀消失

第三步:在虚拟环境中搭建gpu版tensorflow

激活虚拟环境:
source tutorial-env/bin/activate

终端输入:
pip install tensorflow-gpu

屏幕截图.png

PS:提示权限问题,使用root命令:
sudo pip install tensorflow-gpu

由于tensorflow是外网,速度很慢,可以下将终端中提示网址的安装包下载好,放在当前目录下直接安装,


屏幕截图.png

在终端输入安装名字(tap可以补全):
pip install tensorflow_gpu-1.14.0-cp36-cp36m-manylinux1_x86_64.whl

2019-07-15 15-54-05 的屏幕截图.png

第四步:测试虚拟环境

终端输入:
source tutorial-env/bin/activate
进入虚拟环境

输入:python
进入python环境

屏幕截图.png

输入>>>import tensorflow as tf
导入tensorflow库


没有报错,tensorflow安装好了,再继续测试gpu,设置一个常量
输入>>>a = tf.constant(1)

打开tensorflow的对话,输入>>>sess = tf.Sesion()


打印出很多,会出现gpu型号和相关信息

用常量测试一下这个对话>>>sess.run(a)


打印出常量a的值即整个环境搭建好了。

第五步:复用

在下次需要搭建相同环境时,可以用pip将相关的库一次导入。
输入:pip freeze list


可以看到安装的所有的库

输入:pip freeze list > requirements.txt


将开发环境使用的包,导出txt文件中,下次用的时候可以直接安装txt文件中的库。

输入:pip install -r requirements.txt


调用txt文件,直接安装其中的各个库。
PS:请注意路径问题,ls一下,看看相关文件是不是在当前目录下。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,839评论 6 482
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,543评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 153,116评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,371评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,384评论 5 374
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,111评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,416评论 3 400
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,053评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,558评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,007评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,117评论 1 334
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,756评论 4 324
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,324评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,315评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,539评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,578评论 2 355
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,877评论 2 345

推荐阅读更多精彩内容