1. 程式人生 > >nvidia gpu kill程序後,視訊記憶體仍未清

nvidia gpu kill程序後,視訊記憶體仍未清

[email protected]:~$ sudo fuser -v /dev/nvidia*
                     使用者     程序號 許可權   命令
/dev/nvidia0:        root      2750 F...m Xorg
                     xs       14544 F...m python
                     xs       14546 F...m python
                     xs       14547 F...m python
                     xs       14548 F...m python
                     xs       14549 F...m python
                     xs       14550 F...m python
                     xs       14551 F...m python
                     xs       14552 F...m python

殺掉這8個程序號連續的子程序即可


pytorch設定執行緒數為8

dataloaders = {x: torch.utils.data.DataLoader(image_datasets[x], batch_size=opt.batchsize,num_workers=8)

kill掉該程式碼對應的程序後,發現視訊記憶體仍未清除
可以使用sudo fuser -v /dev/nvidia*詳細檢視最近在GPU上執行的程序
其中:
fuser:可以顯示出當前哪個程式在使用磁碟上的某個檔案、掛載點、甚至網路埠,並給出程式程序的詳細資訊
-v:詳細模式
/dev/nvidia*:所有和英偉達相關的介面(如GPU)