版本不对会出问题,如果你版本和我不一样,那么就不要照着做了
Anadonda版本随意
>>conda create --name tensorflow-gpu python=3.5
>>activate tensorflow-gpu
>>pip install tensorflow-gpu==1.4.0
CuDNN记得要v6.0
然后测试
>>import tensorflow
>>sess = tensorflow.Session(config=tensorflow.ConfigProto(log_device_placement=True))
ok,成功了。
查看自己安装的版本:
pip show tensorflow-gpu
pip show keras
nvcc --version