我的测试:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()`
错误:
C:\升\工作\ tensorflow-1.1.0 \ tensorflow \ stream_executor \ CUDA \ cuda_driver.cc:405]
调用cuInit失败:CUDA_ERROR_NO_DEVICE
- >但“/ cpu:0”工作正常
配置:
为什么Tensorflow无法检测到我的GPU?
答案 0 :(得分:8)
问题已解决on GitHub。如果您为CUDA_VISIBLE_DEVICES
环境变量设置了无效值,则会显示此错误消息,例如当您只有一个GPU(标识为0
)并设置CUDA_VISIBLE_DEVICES=1
或CUDA_VISIBLE_DEVICES=2
时。
答案 1 :(得分:5)
真正的答案是添加以下代码行:
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"