windows - TensorFlow: mislykket opkald til cuInit: CUDA\_ERROR\_NO\_DEVICE

Indlæg af Hanne Mølgaard Plasc

Problem



Min test:


import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()`


Fejl:


c: \ l \ arbejde \ tensorflow-1.1.0 \ tensorflow \ stream\_executor \ cuda \ cuda\_driver.cc: 405]]


mislykket opkald til cuInit: CUDA\_ERROR\_NO\_DEVICE


-> men '/cpu: 0' virker fint


Config:


nvidia-smi:
Indtast billedbeskrivelse her [7]



  • CUDA Version 9.1

  • tensorflow-1.1.0

  • Windows 10

  • cudnn64\_7.dll (installeret i C: \ Programmer \ NVIDIA GPU Computing Toolkit \ CUDA \ v9.1 \ bin)

  • Kun tensorflow-gpu er installeret i mit Conda-miljø



Hvorfor kan ikke Tensorflow opdage min GPU?

Bedste reference


Problemet blev løst på GitHub. Denne fejlmeddelelse vises, hvis du angiver en ugyldig værdi for miljøværdien CUDA\_VISIBLE\_DEVICES, f.eks. når du kun har en enkelt GPU (som har ID 0) og indstil CUDA\_VISIBLE\_DEVICES=1 eller CUDA\_VISIBLE\_DEVICES=2. [8]

Andre referencer 1


Det rigtige svar er at tilføje følgende linje kode:


import os
os.environ['CUDA\_VISIBLE\_DEVICES'] = "0"