Невозможно получить доступ к графическому процессору в PyTorch: torch.cuda.is_available() имеет значение False и ошибка ⇐ Python

[Anonymous]

Я работаю над экземпляром EC2 со следующей настройкой:
Версия драйвера NVIDIA: 560.35.03
Версия CUDA (nvidia-smi): 12.6
Версия nvcc: 12.4
Модель графического процессора: NVIDIA A10G
Я установил PyTorch и CUDA с помощью следующей команды:

Код: Выделить всё

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

Однако, когда я запускаю свой код PyTorch, я не могу получить доступ к графическому процессору, и torch.cuda.is_available() возвращает False.Вот вывод nvidia-smi:

Код: Выделить всё

Sun Jan 19 09:02:48 2025
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 560.35.03              Driver Version: 560.35.03      CUDA Version: 12.6     |
|-----------------------------------------+------------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id          Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. |
|                                         |                        |               MIG M. |
|=========================================+========================+======================|
|   0  NVIDIA A10G                    On  |   00000000:00:1E.0 Off |                    0 |
|  0%   23C    P8              8W /  300W |       1MiB /  23028MiB |      0%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+
+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI        PID   Type   Process name                              GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|  No running processes found                                                             |
+-----------------------------------------------------------------------------------------+

Что я пробовал:

• Проверил, что драйвер NVIDIA установлен правильно.
• Подтверждена совместимость CUDA с драйвером и PyTorch.
• Перезапущен машина после установки.
• Установлен pytorch-cuda=12.4 в соответствии с версией nvcc.
• Убедитесь, что pytorch и pytorch-cuda установлены из каналов pytorch и nvidia соответственно.
• Почему PyTorch не может обнаружить графический процессор?
• Как исправить эту ошибку и обеспечить доступность графического процессора для PyTorch?

Будем очень признательны за любые рекомендации. Спасибо!

Подробнее здесь: https://stackoverflow.com/questions/793 ... cuda-error