web-dev-qa-db-ja.com

TensorFlow：cuInitへの失敗した呼び出し：CUDA_ERROR_NO_DEVICE

私のテスト：

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： enter image description here

  • CUDAバージョン9.1
  • tensorflow-1.1.0
  • ウインドウズ10
  • cudnn64_7.dll（C：\ Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\binにインストール）
  • 私のConda環境にtensorflow-gpuのみがインストールされています

TensorflowがGPUを検出できないのはなぜですか？

windowstensorflowcudnn
6
Vincent Roye

問題は解決しました GitHubで 。このエラーメッセージは、CUDA_VISIBLE_DEVICES環境変数に無効な値を設定した場合に表示されます。単一のGPU（ID 0）のみがあり、CUDA_VISIBLE_DEVICES=1またはCUDA_VISIBLE_DEVICES=2を設定している場合。

16
mrry

本当の答えは、次のコード行を追加することです。

import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"
11
fly