在執(zhí)行深度學(xué)習(xí)任務(wù)的時(shí)候，經(jīng)常需要查看GPU的使用情況

具體使用的命令是nvidia-smi

根據(jù)結(jié)果可以看到進(jìn)程2226042在跑訓(xùn)練

執(zhí)行以下命令：

ps -f -p 2226042

其中

• UID 表示用戶ID

• PID 表示進(jìn)程號(hào)

• PPID 表示父進(jìn)程號(hào)

• TIME 表示執(zhí)行時(shí)間

• CMD 表示執(zhí)行命令

查看一臺(tái)服務(wù)器上所有顯卡上的使用者和GPU占用情況

需要安裝gpustat

pip install gpustat

安裝完成后使用下面命令即可

watch -n 2 --color gpustat --c

查看完成后?Crtl +c?即可退出！

實(shí)時(shí)刷新：

watch -n 0.5 -d nvidia-smi

解決多塊GPU指定的問題：

import os

# 重新標(biāo)記第0塊GPU，之后用0即表示第一塊GPU

os.environ['CUDA_VISIBLE_DEVICES'] = '1'

當(dāng)前文件夾下有多少個(gè)文件：（不包括目錄）

ls -l | grep "^-" | wc -l

查看當(dāng)前磁盤空間

df -hl