最近在训练Mars数据集,采用一块GTX1080显卡(8GB内存),发现训练时没问题,测试时出现错误:cuda out of memory。
用nvidia-smi命令查看了下显卡情况,发现除了运行的程序外,桌面X和gnome占了差不多200多M的内存。为了让程序能100%地使用显卡内存,我决定把桌面关了。具体步骤如下:
1.桌面环境下,打开终端
Ctrl+Alt+t
2.关闭桌面
sudo service lightdm stop
此时,屏幕就变黑了。
3.进入命令行界面
Ctrl+Alt+F1
按照提示输入账号、密码,之后会看到和桌面环境下的终端一样的命令行界面。
4.查看当前的显卡使用情况
nvidia-smi
会发现此时没有任何程序使用显卡。
5.用命令行跑程序
python code.py
6.返回桌面环境
sudo service lightdm start