Не работает CUDA из Numba (Python3) на Ubuntu 20.04

Question

Я хочу использовать CUDA из Python3 с помощью библиотеки Numba.

ОС: Ubuntu 20.04

Видеокарта: Nvidia GeForce GTS 450

Используется драйвер Nouveau.

Установил nvidia-cuda-toolkit с помощью sudo apt install nvidia-cuda-toolkit. Проверяю.

>>> from numba import cuda
>>> print(cuda.gpus)

Выдаёт

numba.cuda.cudadrv.error.CudaSupportError: Error at driver init:
[100] Call to cuInit results in CUDA_ERROR_NO_DEVICE:

Пробовал устанавливать драйвер Nvidia с помощью программы "Драйверы" (из графического интерфейса). Установил NVIDIA driver metapackage из nvidia-driver-460 (проприетарное ПО, протестировано). Проверяю.

Python 3.8.5 (default, Jan 27 2021, 15:41:15) 
[GCC 9.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from numba import cuda
>>> print(cuda.gpus)
Traceback (most recent call last):
  File "/home/user1/.local/lib/python3.8/site-packages/numba/cuda/cudadrv/driver.py", line 237, in initialize
    self.cuInit(0)
  File "/home/user1/.local/lib/python3.8/site-packages/numba/cuda/cudadrv/driver.py", line 300, in safe_cuda_api_call
    self._check_error(fname, retcode)
  File "/home/user1/.local/lib/python3.8/site-packages/numba/cuda/cudadrv/driver.py", line 335, in _check_error
    raise CudaAPIError(retcode, msg)
numba.cuda.cudadrv.driver.CudaAPIError: [100] Call to cuInit results in CUDA_ERROR_NO_DEVICE

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/user1/.local/lib/python3.8/site-packages/numba/cuda/cudadrv/devices.py", line 43, in __str__
    return ', '.join([str(d) for d in self.lst])
  File "/home/user1/.local/lib/python3.8/site-packages/numba/cuda/cudadrv/devices.py", line 26, in __getattr__
    numdev = driver.get_device_count()
  File "/home/user1/.local/lib/python3.8/site-packages/numba/cuda/cudadrv/driver.py", line 346, in get_device_count
    self.cuDeviceGetCount(byref(count))
  File "/home/user1/.local/lib/python3.8/site-packages/numba/cuda/cudadrv/driver.py", line 280, in __getattr__
    self.initialize()
  File "/home/user1/.local/lib/python3.8/site-packages/numba/cuda/cudadrv/driver.py", line 240, in initialize
    raise CudaSupportError("Error at driver init: \n%s:" % e)
numba.cuda.cudadrv.error.CudaSupportError: Error at driver init: 
[100] Call to cuInit results in CUDA_ERROR_NO_DEVICE:
>>>

Ничего не изменилось. Установил обратно драйвер Nouveau т.к. тот драйвер (Nvidia) не правильно определял разрешение экрана и делал его очень маленьким. Но это другая, отдельная проблема. Если CUDA с драйвером Nvidia заработает, то я буду решать эту проблему отдельно.

После переустановки, почему-то, nvidia-cuda-toolkit удалился сам по себе. Установил его снова. Проверяю. Происходит в точности такая же ошибка.

Помогите, пожалуйста.

БЛОГ НА HUSL

Не работает CUDA из Numba (Python3) на Ubuntu 20.04

Ответы (0 шт):