安装GPU版本的TensorFlow

标签：blog   download   ogr   post   int   python3   path   blank   type   

• win10 x64
• python 3.6
• 显卡 GTX 940mx
• Cuda 8.0
• cudnn v5.1
• TensorFlow-gpu 1.0.0

1. 安装CUDA

显卡型号支持：https://developer.nvidia.com/cuda-gpus

下载安装CUDA，安装好之后把CUDA安装目录下的bin和lib\64添加到Path环境变量中

2. 安装cuDNN

下载CuDNN需要注册账号

下载安装cuDNN

解压压缩包，把压缩包中bin，include，lib中的文件分别拷贝到c:\Program Files\NVIDIA GPU Computing Tookit\CUDA\v8.0 目录下对应目录中

3. 把c:\Program Files\NVIDIA GPU Computing Tookit\CUDA\v8.0\extras\CUPTI\libx64\cupti64_80.dll拷贝到c:\Program Files\NVIDIA GPU Computing Tookit\CUDA\v8.0\bin

4. 进入cmd，安装gpu版本的TensorFlow

pip install tensorflow-gpu

5. 测试是否安装成功　

import tensorflow as tf
 
x= tf.random_normal((10, 10))
y = tf.random_normal((10, 50))
z = tf.matmul(x, y)
sess = tf.InteractiveSession()
sess.run(z)

参考文献：

【1】【Win7 x64】+【annaconda3】+ 【python3.5.2】+【tensorflow-gpu】 [最终配置 gtx 940mx + Cuda8.0+cudnn v5.1 + tensorflow-gpu1.0.0 ]

安装GPU版本的TensorFlow

标签：blog   download   ogr   post   int   python3   path   blank   type   

原文地址：https://www.cnblogs.com/nxf-rabbit75/p/10636566.html