python3.5安装tensorflow-gpu对应版本_安装GPU版本的TensorFlow

win10 x64

python 3.6

显卡 GTX 940mx

Cuda 8.0

cudnn v5.1

TensorFlow-gpu 1.0.0

1. 安装CUDA

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

2. 安装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)

参考文献：