指定GPU運行python程序


一、命令行運行python程序時

1、首先查看哪些GPU空閑,nvidia-smi顯示當前GPU使用情況。

nvidia-smi 

2、然后指定空閑的GPU運行python程序。

CUDA_VISIBLE_DEVICES=0,2,3 python test.py 

二、在python程序中指定GPU

import os os.environ["CUDA_VISIBLE_DEVICES"] = "0,2,3" 

PS:周期性輸出GPU使用情況

每 10s 顯示一次GPU使用情況:

watch -n 10 nvidia-smi 

   


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM