Theano 和 pygpu:错误

Theano and pygpu: errors

我将 Theano 与 pygpu 一起使用。

一般来说,它运行良好,直到,由于我仍然无法理解的原因,当我尝试导入 theano 时,它显示以下错误:

ERROR (theano.gpuarray): Could not initialize pygpu, support disabled
Traceback (most recent call last):
  File "/home/poko/Software/anaconda2/lib/python2.7/site-packages/theano/gpuarray/__init__.py", line 220, in <module>
    use(config.device)
  File "/home/poko/Software/anaconda2/lib/python2.7/site-packages/theano/gpuarray/__init__.py", line 207, in use
    init_dev(device, preallocate=preallocate)
  File "/home/poko/Software/anaconda2/lib/python2.7/site-packages/theano/gpuarray/__init__.py", line 94, in init_dev
    **args)
  File "pygpu/gpuarray.pyx", line 651, in pygpu.gpuarray.init
  File "pygpu/gpuarray.pyx", line 587, in pygpu.gpuarray.pygpu_init
GpuArrayException: cuInit: CUDA_ERROR_UNKNOWN: unknown error

如果我重新启动我的计算机,它会再次正常运行一段时间(有时几天..)。

现在这种情况很奇怪,因为这些事情要么有效,要么无效。 除了从 nvidia-smi 观察到 xorg 和 chrome 确实占用了大量内存外,我对产生错误的原因一无所知:

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      1332      G   /usr/lib/xorg/Xorg                           392MiB |
|    0      2243      G   cinnamon                                     110MiB |
|    0      4927      G   ...-token=39C210A3DFA14C5D81FA629C813B843D   154MiB |
+-----------------------------------------------------------------------------+

事实证明,我可以通过卸载 nvidia_uvm 模块来消除错误,方法是:

sudo rmmod nvidia_uvm

之后会自动重新加载

希望这有助于其他人遇到该问题。