安装GPU版本的TensorFlow

Posted 2021-11-25 nxf-rabbit75

• 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 ]