如何指定GPU執行程式

2021-09-17 08:45:09 字數 562 閱讀 7133

連伺服器有時候會出現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_devices=6(或cuda_visible_devices=6,7)command

舉例說明:

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

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

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

則執行命令修改為:

cuda_visible_devices=5,6,7 python train.py

cuda_visible_devices=1,2 python mytrain_vgg.py

指定GPU執行程式

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 comm...

怎樣指定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...

Python 如何執行程式

1.python 直譯器 直譯器是一種讓其他程式執行起來的程式。當你寫一段 python 程式,python 直譯器將讀取程式,並按照其中的命令執行,得出結果。實際上,直譯器是 與機器的計算機硬體這間的軟體邏輯層。2.python 位元組碼 python內部 對大多數使用者是完全隱藏的 會先將源 編...