参考的https://blog.youkuaiyun.com/chicunlian4601/article/details/89919173这个博客。
1.https://docs.microsoft.com/en-us/visualstudio/productinfo/2017-redistribution-vs#vs2017-download下载VS2017
2.https://www.geforce.cn/drivers/results/138113下载显卡驱动。
3.https://developer.nvidia.com/cuda-toolkit-archive下载cuda10.0。
4.https://developer.nvidia.com/rdp/cudnn-archive下载cudnn7.3.1,然后解压,把文件放到默认路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0下面。
5.conda创建一个环境
conda create -n tensorflow-gpu python=3.6
6.激活tensorflow-gpu环境,安装tensorflow
pip install tensorflow-gpu==1.13.1
7.测试
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
如果出现显卡型号,就成功了!
采坑笔记
第一次安装,直接安装cuda10.0,出现了DLL not found的问题。然后卸载了NVIDIA的相关软件,先安装的驱动(第2步),后安装cuda10.0。然后就好了。
还有一个问题numpy版本过高的问题,降到1.16.0就可以了。