Windows配置GPU环境:
- 安装anaconda: 链接:https://pan.baidu.com/s/1RFWfA7mDaxxi31rf3XCC3g 提取码:as1x
- 安装Visual Studio 2017: https://visualstudio.microsoft.com/zh-hans/downloads/
- 安装cuda: 链接:https://pan.baidu.com/s/1WID6vH9N7N_4s_hP1QQ6qg 提取码:fqvg
- 将cudnn文件夹里的内容拷贝到CUDA的安装目录并覆盖相应的文件夹
- 打开cmd,输入命令:pip install tensorflow-gpu==1.8
- 测试:打开cmd,输入python
- 然后输入import tensorflow as tf ,没有报错说明安装成功
若没有GPU,省略2,3,4步骤。第五步改为:pip install tensorflow==1.8
Ubuntu配置GPU环境:
Ubuntu配置过程比较复杂,主要参考链接:
https://blog.csdn.net/u010159842/article/details/54344683