新手安装anaconda并配置gpu版本的tensorflow,菜鸟成功篇
新手安装anaconda并配置gpu版本的tensorflow,菜鸟成功篇
一. 看GPU支持的cuda版本及对应的cuDNN版本,下载cuda在线安装的版本(9.0以下才支持tensorflow)
二. 安装
其它组件也可以选择不安装。
下一步,可以自己修改安装路径,默认路径如下:
等待安装好
三. 下载的各种补丁,如果修改了路径就更改安装在同一路径下,不装貌似没啥大影响。
四. 下载cuDNN 7.0,官网下载需要注册账号,挺麻烦,可以自行从网上其他地方找。下载后无需安装,直接解压得到三个文件夹。把这三个文件夹里的文件拷贝到CUDA Toolkit 9.0安装目录下相同名字的文件夹中,
五. cmd下 键入nvcc -V出现NVIDIA信息一般就安装好了,有很多其它测试方法
六. Anaconda的安装很简单,安装时加入环境等两个选项都选就好
七. cuda环境变量配置更改
八. Conda加入系统环境变量
看网上有说要添加三个变量的:
九. 然后打开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运行情况
参考了特别多别的博客,,,非常感谢分享,,不一一列出了,