pytorch預設使用從0開始的gpu，如果指定的gpu正在執行程式，需要指定其他gpu。目前實驗發現指定的gpu不存在時不會報錯，系統自動尋找可用的gpu。
指定方式
1
11. 直接在終端設定
cuda_visible_devices=0 python3 trainval_net.py
指定了使用0號gpu
2
22.在程式中指定
import os
os.environ[「cuda_visible_devices」] = 「1」
指定了1號gpu
3
33.使用函式 set_device
import torch
torch.cuda.set_device(id)
不過官方建議使用cuda_visible_devices，不建議使用 set_device 函式。
tensorflow中使用指定的gpu及gpu視訊記憶體可參考：
PyTorch中使用指定的GPU
pytorch預設使用從0開始的gpu，如果gpu0正在執行程式，需要指定其他gpu。有如下兩種方法來指定需要使用的gpu。1.類似tensorflow指定gpu的方式，使用cuda visible devices。1.1 直接終端中設定 cuda visible devices 1 python ...
PyTorch中使用指定的GPU
pytorch預設使用從0開始的gpu，如果gpu0正在執行程式，需要指定其他gpu。有如下兩種方法來指定需要使用的gpu。1.類似tensorflow指定gpu的方式，使用cuda visible devices。1.1 直接終端中設定 cuda visible devices 1 python ...
使用指定GPU跑Pytorch
有如下兩種方法來指定需要使用的gpu。類似tensorflow指定gpu的方式，使用cuda visible devices。1.1 直接終端中設定 cuda visible devices 1 python my script.py 1.2 python 中設定 import os os.envi...