tensorflow-gpu+"Failed to create session"

成因: 未给系统指定相应使用的GPU

解决:

  1. 层面1: 针对各个虚拟环境:
    修改~/.virtualenv/YOUR_VENV/bin/postactivate, 在末尾添加行export CUDA_VISIBLE_DEVICES="1"即可.

  2. 层面2: 针对系统用户:
    同上修改~/.bashrc即可.

文章来自:http://www.cnblogs.com/ZhengPeng7/p/7833352.html
© 2021 jiaocheng.bubufx.com  联系我们
ICP备案:鲁ICP备09046678号-3