- 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需要注册账号
解压压缩包,把压缩包中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)
参考文献:
原文地址:https://www.cnblogs.com/nxf-rabbit75/p/10636566.html
时间: 2024-10-30 22:37:07