Как удалить cuInit не удалось: неизвестная ошибка в CUDA (PyCuda)

Question

У меня установлена ​​CUDA, затем я попробовал следующее:

In [1]: import pycuda.driver as cuda

In [2]: cuda.init()
---------------------------------------------------------------------------
Error                                     Traceback (most recent call last)
<ipython-input-2-2845c9c0ab3c> in <module>()
----> 1 cuda.init()

Error: cuInit failed: unknown error

Как можно устранить ошибку?

У меня уже установлено это:

$ which nvidia-modprobe
/usr/bin/nvidia-modprobe

Моя другая информация:

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2015 NVIDIA Corporation
Built on Tue_Aug_11_14:27:32_CDT_2015
Cuda compilation tools, release 7.5, V7.5.17

А это:

$ uname -a
Linux foobar1 4.10.0-28-generic #32~16.04.2-Ubuntu SMP Thu Jul 20 10:19:48 UTC 2017 x86_64 x86_64 x86_64 GNU/Linux

А это информация для nvidia-smi

$ nvidia-smi
Mon Nov 19 15:56:43 2018
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 410.48                 Driver Version: 410.48                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  Quadro P400         Off  | 00000000:AF:00.0 Off |                  N/A |
| 34%   45C    P0    N/A /  N/A |      0MiB /  1999MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

Answer 1

cuInit failed: unknown error часто является результатом того, что модуль ядра nvidia-uvm не загружается.Я периодически сталкивался с этой проблемой в Ubuntu.

sudo nvidia-modprobe -u должен решить эту проблему.То есть до перезагрузки.Затем вам нужно будет сделать это снова.

Еще один обходной путь - запустить ваше сбойное приложение как root один раз .УДОВЛЕТВОРЯЕТ это в этом случае среда выполнения CUDA попытается загрузить отсутствующий модуль (и, скорее всего, преуспеет в этом, потому что мы работаем от имени пользователя root).