pytorch на ubuntu runteimeError: ошибка CUDA: все устройства, способные CUDA заняты или недоступны
Я попытался построить Pytorch от источника, как рекомендовал.
Спецификации: GPU: GT 710, водитель 460.32.03, CUDA 11.2, Python 3.8.5.
Тест:
import torch
print(torch.__version__) -- 1.7.1
print(torch.cuda.is_available()) -- True
print(torch.backends.cudnn.enabled) -- True
device = torch.device('cuda')
print(torch.cuda.get_device_properties(device)) -- _CudaDeviceProperties(name='GeForce GT 710', major=3, minor=5, total_memory=1998MB, multi_processor_count=1)
print(torch.tensor([1.0, 2.0]).cuda()) --RuntimeError: CUDA error: all CUDA-capable devices are busy or unavailable
2 ответа
Я думаю, что ваша проблема связана с версией CUDA CC, которая устанавливается с вашей картой. От того, что я знаю, вы должны составить от источника и убедиться, что CUDA CC совместима с версией Pytorch. Пожалуйста, обратитесь к https://github.com/pytorch/pytorch/issues/31285
Что похоже, что все в порядке:
• CUDA 11.2, uses Linux Driver >=460.27.03, You have 460.32.03
• Major + Minor Values = 3.5, need at least 3 computational power to use Cuda 11.2
• Python Version is 3.8.5 recommended is => 3.6
You used "built from source install"
Я не думаю, что это проблема, ориентированная на совместимость по большей части (хотя некоторые GT 710 сообщалось, что при вычислительной мощности 3, что не так) Любые размышления о совместимости можно найти здесь «https://docs.nvidia.com/deploy/cuda-compatibility/index.html#source-compatibility», есть команда запроса для двойного проверки проверки CUDA на вашем графическом процессе: это должно быть Расположен в вашей папке установки CUDA в разделе «Дополнительно».
Используйте команду nvidia-smi, чтобы увидеть, какие процессы используют графическую карту и показать результат. Также я бы также порекомендовал запустить запрос вышеупомянутого устройства и публиковать изображение.
Если на видеокарте уже есть процессы, вы можете убить их
NVIDIA-SMI | Greep 'Python' | awk '{print $ 3}' | xargs -n1 kill -9 Где «Python» будет название процесса