简易安装Tensorflow GPU(使用Anaconda 中的conda 命令安装)

1、安装Anaconda

官网下载会比较缓慢,建议从 清华大学镜像站 下载相应版本的Anaconda,本文中使用到的版本为 Anaconda3-5.3.1。

2、修改conda的国内镜像源

由于安装tensorflow 时需要安装大量的依赖包,为了提高安装速度,建议将anaconda conda 切换为国内源。
首先,打开Anaconda Prompt,进入命令行模式(linux直接打开命令行即可)。

1575882815(1).png

命令行中直接使用以下命令:

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

3、使用conda 命令创建新的环境

conda create -n tf-gpu

tf-gpu 是你所创建的环境名称,可以自行修改。

4、安装Tensorflow GPU版本

首先,需要切换上一步中建立的环境:

activate tf-gpu

需要注意的是,Linux环境中切换环境的命令为:

source activate tf-gpu

再通过conda命令安装Tensorflow GPU版本:

conda install tensorflow-gpu

安装过程中输入y就可完成安装,耐心等待即可。

5、解决Jupyer Notebook中import tensorflow报错:[No module named tensorflow]

安装tensorflow,在Anaconda Prompt命令行下的python中可以import tensorflow,但是到了Jupyer Notebook中,执行

import tensorflow as tf

会发生错误,显示报错信息为:No module named tensorflow
为了解决这个问题,需要安装 nb_conda
首先,需要切换上一步中建立的环境,如已在该环境中则跳过此步骤:

activate tf-gpu

安装nb_conda

conda install nb_conda

安装完成后,在命令行中输入:

jupyter notebook

启动jupyter notebook后,在创建文件的选项中就可以看到我们建立的环境了,选择自己的环境创建新文件,import tensorflow as tf 就不会再报错了。

1.png

6、测试tensorflow GPU版本是否能正常调用显卡计算,并解决cuDNN版本对不上问题。

使用conda安装tensorflow GPU版本会自动下载依赖的CUDA以及cuDnn,但是你的环境中以前没有装过,所以默认给你装最新的CUDA10.0,所以需要将NVIDIA显卡的驱动升级到对应的版本。
可以从 NVIDIA驱动官网 下载所对应的显卡驱动,本文以笔记本的GTX 1060显卡为例:

1575897542(1).png

下载最新的驱动安装,即可解决cuDNN版本对不上问题。
测试GPU是否能调用的代码如下:

import tensorflow as tf
import os

os.environ['TF_CPP_MIN_LOG_LEVEL']='2'

a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)

print('GPU:', tf.test.is_gpu_available())

成功的输出如下:

tf.Tensor(3.0, shape=(), dtype=float32)
GPU: True

7、解决spyder相关报错。

打开spyder,点击Tools > PYTHONPATH manager

1575893733(1).jpg

然后依次将Anaconda 中所创建环境的路径添加进去,并且添加基础的路径。需要添加的路径如图所示:
其中创建环境的路径为:
安装Anaconda的路径\envs\环境名\Lib
安装Anaconda的路径\envs\环境名\Lib\site-packages
安装Anaconda的路径\envs\环境名\Scripts

以本文为例,创建环境的路径为:

C:\kaifa\anaconda3-5-1\envs\tf-gpu\Lib
C:\kaifa\anaconda3-5-1\envs\tf-gpu\Lib\site-packages
C:\kaifa\anaconda3-5-1\envs\tf-gpu\Scripts

其中基础路径为:
安装Anaconda的路径\Lib
安装Anaconda的路径\Lib\site-packages
安装Anaconda的路径\Scripts

以本文为例,基础的路径为:

C:\kaifa\anaconda3-5-1\Lib
C:\kaifa\anaconda3-5-1\Lib\site-packages
C:\kaifa\anaconda3-5-1\Scripts

需要注意的是,一定要将所创建环境的路径排在前面,不然在调用GPU运算时出错。
添加好路径后,点击Synchronize按钮确定即可。

1575895172(1).png

注意:由于修改python编译器环境,需要以"Anaconda Prompt"的命令行模式进入spyder,具体操作如下:
进入创建的环境后,输入spyder,即可打开spyder。

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