tensorflow-gpu运行的问题

    xiaoxiao2024-12-22  4

    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的文件中加入下面两句就能成功运行了

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

    然后就成功运行了

    最新回复(0)