Я запускаю PyTorch в Linux.
ОС — Ubuntu 22.04.1. Графический процессор — NVIDIA RTX A4000. CUDA версии 11.5.
Кроме того, я использую версию PyTorch CUDA 11.7.
Когда я запускаю Python3 на терминале, PyTorch не может обнаружить CUDA. Я попытался перезагрузить, но проблема не устранена.
Python 3.10.6 (main, Nov 14 2022, 16:10:14) [GCC 11.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
[PATH]/.local/lib/python3.10/site-packages/torch/cuda/__init__.py:88: UserWarning: CUDA initialization: Unexpected error from cudaGetDeviceCount(). Did you run some cuda functions before calling NumCudaDevices() that might have already set an error? Error 803: system has unsupported display driver / cuda driver combination (Triggered internally at ../c10/cuda/CUDAFunctions.cpp:109.)
return torch._C._cuda_getDeviceCount() > 0
False
После обновления драйвера CUDA до версии 11.7 PyTorch успешно распознает CUDA.
обновите до последней версии драйвер для вашего графического процессора.