我的测试：
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()`字符串
错误：
c：\l\work\tensorflow-1.1.0\tensorflow\stream_executor\cuda\cuda_driver.cc：405]
调用cuInit失败：CUDA_ERROR_NO_DEVICE
->但“/cpu：0”工作正常
配置：
nvidia-smi：
- CUDA 9.1版
- tensorflow-1.1.0
- Windows 10
- cudnn64_7.dll（安装在C：\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\bin中）
- 我的Conda环境中只安装了tensorflow-gpu
为什么Tensorflow无法检测我的GPU？
3条答案
按热度按时间doinxwow1#
如果您为
CUDA_VISIBLE_DEVICES环境变量设置了无效的值，例如当您只有一个GPU（其ID为0）并设置了CUDA_VISIBLE_DEVICES=1或CUDA_VISIBLE_DEVICES=2时，将显示此错误消息。bxgwgixi2#
真正的答案是添加以下代码行：
字符串
dz6r00yl3#
如果您使用的是Google Colab，则可能是您的运行时不是GPU。单击“更改”菜单。选择“更改运行时类型”。现在选择GPU。