Theano для GPU ошибка памяти - PullRequest
0 голосов
/ 06 июня 2018

Несколько часов назад я успешно запустил и запустил среду графического процессора Theano + Keras.Я даже протестировал некоторый код, чтобы убедиться, что он выполняется на GPU.Однако, когда я запускаю import theano сейчас, я получаю следующую ошибку:

ERROR (theano.gpuarray): Could not initialize pygpu, support disabled Traceback (most recent call last): . . . . GpuArrayException: cuDevicePrimaryCtxRetain: CUDA_ERROR_OUT_OF_MEMORY: out of memory

Я использую графический процессор на нашем университетском сервере, и он используется многими студентами в лаборатории.Возможно, ошибка связана с нехваткой памяти из-за других запущенных процессов?Вывод nvidia-smi показан ниже.Процесс с PID 29586 - мой.

+-----------------------------------------------------------------------------+ | NVIDIA-SMI 387.12 Driver Version: 387.12 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | |===============================+======================+======================| | 0 GeForce GTX TIT... Off | 00000000:01:00.0 On | N/A | | 48% 83C P2 196W / 250W | 12152MiB / 12198MiB | 99% Default | +-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0     10977      C   python                                      5506MiB |
|    0     24129      C   python                                      6323MiB |
|    0     25238      G   /usr/lib/xorg/Xorg                           110MiB |
|    0     25773      G   /usr/bin/gnome-shell                          90MiB |
|    0     29586      C   python                                       106MiB |
+-----------------------------------------------------------------------------+

Графический процессор - это Nvidia Titan X. Я много гуглял эту ошибку и перепробовал множество методов за последние несколькоч.Пожалуйста, помогите.

1 Ответ

0 голосов
/ 06 июня 2018

Для простоты, да, на карте недостаточно памяти.TITAN X имеет 12 ГБ оперативной памяти, и первые процессы почти все используют.Возможно, спросите владельца, могут ли они остановить свой процесс или использовать пакет меньшего размера, если они используют его для глубокого обучения.

...