Ubuntu-fr

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