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可在运行时这样写。
怎么指定 GPU
最新推荐文章于 2024-07-26 17:23:36 发布