#1 Le 15/09/2022, à 11:46
- anass783
GPU non disponible avec tensorflow et Torch sous Ubuntu 18.04
Bonjour,
je fais face à un problème lié à un mon GPU. Il n'est pas disponible avec Tensorflow ni avec Torch.
import tensorflow as tf
device_name = tf.test.gpu_device_name()
if device_name != '/device:GPU:0':
print('The device name is: ',device_name)
--> print : The device name is: '' # the device_name returned is empty
else:
print("GPU available")
La version de CUDA installé est :
$ nvcc --version
--> nvcc: NVIDIA (R) Cuda compiler driver
--> Copyright (c) 2005-2022 NVIDIA Corporation
--> Built on Tue_May__3_18:49:52_PDT_2022
--> Cuda compilation tools, release 11.7, V11.7.64
--> Build cuda_11.7.r11.7/compiler.31294372_0
La version de NVIDIA est :
nvidia-smi
--> NVIDIA-SMI 515.65.01 Driver Version: 515.65.01 CUDA Version: 11.7
La version de Tensorflow-gpu est :
import tensorflow as tf
print(tf.__version__)
--> 2.0.0
J'ai lu que la version 11.7 de CUDA peut correspondre à la version 515.65.01 de NVIDIA pourtant ca ne marche pas.
Quelqu'un peut m'aider svp ? Merci
Hors ligne