踩了一下午的坑,被大佬几句话就给点明白了
我的电脑联想win10.显卡gtx1050,cuda9.0,cudnn7.4.1.5,TensorFlow-GPU1.8.0,anaconda4.4.10(这个应该关系不大)之前主要是安装的这些东西不是配套的,所以如果大家安装失败了可能只是安装的版本不对,不需要改一大堆配置,可能重新安装个别的版本就能成功了,
CUDA9.0 : https://developer.nvidia.com/cuda-90-download-archive
CUDNN:https://developer.nvidia.com/cudnn
tensorflow-gpu:pip install tensorflow-gpu=1.8.0
cudnn下载解压缩后直接放到cuda的目录下和bin、lib、include合并就好,cuda环境变量安装后自动配置成功,不需要改什么
然后就是网上说的
conda create -n tensorflow-gpu python=3.6
activate tensorflow-gpu
pip install tensorflow-gpu=1.8.0
python
import tensorflow as tf
如果这个导入成功了,我感觉基本有戏。
纪念一下运行成功的截图