CUDA_ERROR_NO_DEVICE如何避免每次设置os.environ ['CUDA_VISIBLE_DEVICE'] ='0'

时间:2018-12-05 11:21:44

标签: python tensorflow cuda nvidia ubuntu-18.04

在将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

0 个答案:

没有答案