Недоступна CUDA на платформе GPU
Статья создана
Обновлена 16 августа 2024 г.
Описание проблемы
CUDA недоступна на платформе GPU.
Решение
Необходимо выполнить команды:
sudo apt update -y.sudo apt upgrade -y.pip3 install torch.
После этого нужно проверить доступность CUDA: torch.cuda.is_available().
Альтернативой может стать вариант использовать сервис DataSphere, где используются конфигурации на базе GPU. У каждого проекта сервиса есть несколько Docker-образов с определенным набором ПО. Если этого будет недостаточно, то можно собрать собственный образ для проекта.
Примечание
CUDA и cuDNN уже встроены в базовый Docker-образ.
Проверить версию этих компонентов можно двумя командами:
- Проверить версию CUDA:
!nvcc --version; - Проверить версию cuDNN:
!cat /usr/include/cudnn_version.h.
Дополнительно можно узнать:
- Информацию по NVIDIA:
!nvidia-smi; - Доступность CUDA через torch:
torch.cuda.is_available().