目錄
一、前言
1.隨着深度學習的快速發展,GPU的優劣也決定模型訓練速度的快慢,而瞭解GPU的使用情況可以讓我們合理的分配GPU。
2.現在普遍的深度學習的GPU採用的都是NVIDIA顯卡,而企業也提供了專門的方式供我們去查看GPU的使用情況。對於一些windows10系統的用戶,他們可以在任務管理器的性能界面查看到GPU的使用情況,但是有些用戶就不可以,例如博主(>_<):
二、解決方法
1.我們首先得知道如何查看,其實官方提供的是一個exe文件,叫nvidia-smi.exe,它在
C:\Program Files\NVIDIA Corporation\NVSMI
這個文件夾裏,內容如下:
2.我們把這個文件路徑加入到環境變量中, 對PATH變量進行編輯,會進入到PATH的列表,再點擊添加,將上面的路徑填寫進去,例如:
3.然後,我們可以打開cmd,輸入指令:
nvidia-smi
就可以查看GPU的使用情況
其中這個40MiB / 2048MiB就是使用顯存佔總顯存的比例
下方的Processes:就是顯示程序的GPU Memory的使用,現在沒有就顯示No running processes found
這樣的方法是不是很方便呢~