windows10 + 64位python3.6.5 + tensorflow-gpu + GTX 950M
这是我写的第一篇文章,大家多多包涵
- 由于tensorflow的cpu版本实在太慢,cpu太烫。狠下心, 最后终于安装好gpu版本,大家可以参考一下,希望大家可以少踩坑。
- 这个过程python3.5也可以参考,过程差不多的,前提必须是64位的。
- 话不多说先上成功的图
注意点
- tensorflow-gpu版本是有对应的CUDA的,必须下载对。可以先百度找找要安装的版本对应的CUDA
- tensorflow-gpu1.9.0 对应的CUDA版本为V9.0
下面说正事
1.检查显卡型号是否支持cuda tookit,可在NVIDIA的网站上查询 https://developer.nvidia.com/cuda-gpus
2. 安装tensorflow-gpu包时,检查核对tensorflow-gpu版本所要求的CUDA版本号,否则依然回报错误异常
本次安装的tensorflow-gpu1.9.0 对应的CUDA版本为V9.0 ,下载:https://developer.nvidia.com/cuda-toolkit-archive
以及下载CUDAV9.0对应的CUDNN版本(须注册下载): https://developer.nvidia.com/rdp/cudnn-download
下载后就是安装CUDA,双击安装默认那些就好
安装好后就是在系统环境变量里面添加(默认安装的话就是这个路径)
也可以自己右键查看安装的CUDA文件所在路径
打开文件所在的路径, 将对应的CUDNN解压到的内容直接复制到CUDA的文件目录中
- 我的就是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
完成。测试代码,没报错就好了
import tensorflow as tf
a = tf.constant(0.1, dtype=tf.float32)
with tf.Session() as sess:
sess.run(a)