在末尾添加行export CUDA_VISIBLE_DEVICES="1"

tensorflow-gpu+"Failed to create session"

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

  2. 层面2: 针对系统用户:
    同上修改~/.bashrc在末尾添加行export CUDA_VISIBLE_DEVICES="1"

 

你可能感兴趣的:(linux)