怎麼指定 GPU

os.environ[“CUDA_VISIBLE_DEVICES“]=INDEX index用GPUid代替id從前0開始,也可在python文件中指定用os.environ[“CUDA_DEVICE_ORDER“]=“PCI_BUS_ID“] os.environ[“-,其中index用可以用 nvidia-smi查看比如0,CUDA_VISIBLE_DEVICES=index python xxx.py,使用cuda可以指定可用的GPU可在運行時這樣寫。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章