tensorflow-gpu运行的问题

win10系统,anaconda
将python3.7转换成python3.6
在anaconda prompt里面输入以下命令:

conda create -n py35 python=3.6 anaconda

再激活py36的环境

activate py36

就可以使用了
下次在使用时,可以

conda info --envs #查看anaconda里有哪些环境
activate py36#进入py36环境

之前在python3.7的环境下,只能安装tensorflow-1.13及以上版本。因为之前的tensorflow版本只支持python3.5,python3.6,在python3.6下就可以安装tensorflow-gpu==1.8.0了

pip install tensorflow-gpu==1.8.0

运行程序还是发生错误,后来在
train_image_classifier.py的文件中加入下面两句就能成功运行了
tensorflow-gpu运行的问题_第1张图片

config=tf.ConfigProto(allow_soft_placement=True)
...
session_config=config

然后就成功运行了

你可能感兴趣的:(tensorflow-gpu运行的问题)