在将Tensorflow的GPU版本导入为tf之后,当我尝试调用tf.Session()时遇到了这个错误
2018-12-05 12:11:58.995627: E tensorflow/stream_executor/cuda/cuda_driver.cc:300] failed call to cuInit: CUDA_ERROR_NO_DEVICE: no CUDA-capable device is detected
我可以通过在调用tf.Session()之前包含以下代码来绕过它:
import os
os.environ["CUDA_VISIBLE_DEVICES"]="0"
我想知道如何停止每次添加这两行代码并将系统CUDA_VISIBLE_DEVICES设置为0