tensorflow-gpu+"Failed to create session"

Posted 默盒

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了tensorflow-gpu+"Failed to create session"相关的知识，希望对你有一定的参考价值。

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

解决:

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

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

以上是关于tensorflow-gpu+"Failed to create session"的主要内容，如果未能解决你的问题，请参考以下文章