在跑深度学习的时候出现问题
查看GPU使用情况
Check failed: error == cudaSuccess (2 vs. 0) out of memory
修改batch_size的大小错误依旧存在,最后发现是一些进程在占用GPU所导致。关闭这些进程,错误消失。
查看GPU使用情况
nvidia-smi
显示如下:
aicrobo@ubuntu:~$ nvidia-smi
Thu Mar 29 19:52:48 2018
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 367.48 Driver Version: 367.48 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+=