新手安装anaconda并配置gpu版本的tensorflow,菜鸟成功篇

新手安装anaconda并配置gpu版本的tensorflow,菜鸟成功篇

一. 看GPU支持的cuda版本及对应的cuDNN版本,下载cuda在线安装的版本(9.0以下才支持tensorflow)

二. 安装

新手安装anaconda并配置gpu版本的tensorflow,菜鸟成功篇
其它组件也可以选择不安装。

下一步,可以自己修改安装路径,默认路径如下:
新手安装anaconda并配置gpu版本的tensorflow,菜鸟成功篇
等待安装好

三. 下载的各种补丁,如果修改了路径就更改安装在同一路径下,不装貌似没啥大影响。

四. 下载cuDNN 7.0,官网下载需要注册账号,挺麻烦,可以自行从网上其他地方找。下载后无需安装,直接解压得到三个文件夹。把这三个文件夹里的文件拷贝到CUDA Toolkit 9.0安装目录下相同名字的文件夹中,

五. cmd下 键入nvcc -V出现NVIDIA信息一般就安装好了,有很多其它测试方法

六. Anaconda的安装很简单,安装时加入环境等两个选项都选就好

七. cuda环境变量配置更改

新手安装anaconda并配置gpu版本的tensorflow,菜鸟成功篇
新手安装anaconda并配置gpu版本的tensorflow,菜鸟成功篇

八. Conda加入系统环境变量
新手安装anaconda并配置gpu版本的tensorflow,菜鸟成功篇
看网上有说要添加三个变量的:
新手安装anaconda并配置gpu版本的tensorflow,菜鸟成功篇
九. 然后打开anaconda prompt,

conda install
python=3.6降低版本(如果本身装的低版本不需要这一步)

十. 输入如下:

conda create -n tf python=3.6

conda activate tf

pip install tensorflow-gpu==1.5

等待安装完成

(如果不介意直接在base环境下安装即可)

十一 测试,可以打开spyder测试,or直接测试

import tensorflow as tf

sess = tf.Session()

a = tf.constant(1)

b = tf.constant(2)

print(sess.run(a+b))

同时可以打开任务管理器看GPU运行情况

参考了特别多别的博客,,,非常感谢分享,,不一一列出了,