Tensorflow не обнаруживает графический процессор на RTX 3060 Ti

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

tf.config.list_physical_devices('GPU')

Выполнение приведенного выше кода дает мне следующий результат:

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 932414320379148726]

[]

Я использую RTX 3060 Ti вместе с CUDA 11.1 и cudnn 8+ на Python 3.8.5. Я пробовал tensorlfow-gpu (2.3, 2.4 и 2.5 dev), но ни один из GPU не обнаружил никакого решения ???.


python tensorflow2.0 tensorflow2.x
person anim esh    schedule 16.12.2020    source источник

Ответы (1)


arrow_upward
0
arrow_downward

С тензорным потоком вы должны сделать:

import tensorflow as tf

# Get the GPU device name.
device_name = tf.test.gpu_device_name()

# The device name should look like the following:
if device_name == '/device:GPU:0':
    print('Found GPU at: {}'.format(device_name))
else:
    raise SystemError('GPU device not found')

Была ли у вас такая же ошибка при использовании pytorch?

import torch
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print(device)
person pafechet    schedule 16.12.2020
comment
Я изменил cuda на 11.0, и это сработало. - person anim esh; 16.12.2020