指定GPU執行程式

2021-09-28 14:56:11 字數 382 閱讀 5799

1.檢視gpu

nvidia-smi -l

檢視7號gpu

nvidia-smi -q -i 7

如果只看memory情況,可以用:

nvidia-smi -q -i 7 -d memory

設定gpuid

cuda_visible_devices=6(或cuda_visible_devices=6,7)command

這條命令適用於命令列執行tensorflow程式的時候,指定gpu.

只需要在命令之前設定環境變數,簡單來說比如原本程式是命令列執行pythontrain.py

假定這裡gpu總共有八塊,通過nvidia-smi檢視發現5,6,7是空閒的(從0開始編號)

則執行命令修改為:

如何指定GPU執行程式

連伺服器有時候會出現out of memory 解決方法 換用空閒的gpu 檢視gpu nvidia smi l 3.檢視7號gpu nvidia smi q i 7 如果只看memory情況。可以用 nvidia smi q i 7 d memory 3.設定gpuid cuda visible ...

怎樣指定GPU執行程式

連伺服器有時候會出現out of memory 解決方法 換用空閒的gpu 1.檢視gpu nvidia smi l 2.檢視7號gpu nvidia smi q i 7 如果只看memory情況。可以用 nvidia smi q i 7 d memory 3.設定gpuid cuda visibl...

Powershell 以指定的賬戶執行程式

很多的情況下,需要以指定的賬戶執行程式 例如在非管理員賬戶下要以管理員許可權執行程式,就可以用下面的ps指令碼 1 2 3 4 5 6 7 8 script editor snail yu date 2014 09 17 uname administrator 指定使用者名稱,administrat...