pycharm設置多GPU運行


轉載請注明出處:

http://www.cnblogs.com/darkknightzh/p/8576825.html

 

如果使用多gpu運行程序,可以直接使用CUDA_VISIBLE_DEVICES=0,1,2,3 python xxx.py來設置該程序可見的gpu。當然也可以在程序開頭設置os.environ["CUDA_VISIBLE_DEVICES"] = '0,1,2,3'來使用。

 

如果在pycharm中調試時,使用多gpu的話,除了直接在程序中設置os.environ["CUDA_VISIBLE_DEVICES"] = '0,1,2,3',也可以通過下面的方法:

1. 打開Run/Debug Configurations,在Environment variables右邊點擊…處。

2. 點擊下圖中1處的’+’,會出現2處,增加CUDA_VISIBLE_DEVICES,再點擊3處,填入0,1,2,3。(下圖中第一行已經增加了,只不過為了說明,因而這樣表示)

3. 在pycharm中正常運行程序,就可以了(為了測試是否正確,實際上使用的是2,3,4,5這4個GPU,顯示的正確,說明這樣配置有效果)。


免責聲明!

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



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