1. 程式人生 > >伺服器上GPU的相關操作

伺服器上GPU的相關操作

1.用JupyterNotebook跑程式時,當發現某個包沒有匯入時,比如發現tqdm沒有匯入時,用下面的步驟匯入:

(1)點選Running:

(2)點選terminals2,進入下面的頁面:進行pip install tqdm即可

2.檢視伺服器上GPU的佔用情況:

在上面展示的頁面上輸入:nvidia-smi

結果如上圖所示

3.程式中指定使用的GPU節點區域:

os.environ["CUDA_VISIBLE_DEVICES"] = "5"#指定節點上面跑程式的區域