1 安装环境
- 系统:Window10 64bit企业版
- 显卡:GeForce GTX1070
- Python: 3.7
- Anaconda: Anaconda3-2019.03-Windows-x86_64
- CUDA: cuda_10.1.168_425.25_win10
- cuDNN: cudnn-10.1-windows10-x64-v7.6.1.34
2 下载软件
Anaconda: Anaconda3-2019.03-Windows-x86_64.exe
CUDA: cuda_10.1.168_425.25_win10.exe
cuDNN: cudnn-10.1-windows10-x64-v7.6.1.34.zip
3 安装流程
1、安装Anaconda
按照指引安装Anaconda,安装过程中需要注意将Anaconda加入PATH
2、安装CUDA
双击exe文件按照指引安装即可
3、安装cuDNN
解压cudnn-10.1-windows10-x64-v7.6.1.34.zip,将include,lib和bin三个文件夹下的文件拷到CUDA安装时对应的文件夹下面即可
4、安装tensorflow-gpu
打开cmd,运行下面的命令,会自动安装符合当前python3.7的gpu版本
pip install tensorflow-gpu
4 验证安装
进入python环境,执行下面代码
import tensorflow as tf
print(tf.test.gpu_device_name())