PyTorch 中如何指定GPU

pytorch預設使用從0開始的gpu，如果gpu0正在執行程式，需要指定其他gpu。

有如下兩種方法來指定需要使用的gpu。

1. 類似tensorflow指定gpu的方式，使用cuda_visible_devices。

1.1 直接終端中設定：

cuda_visible_devices=1 python my_script.py

## 建議使用該方法

import os 
os.environ["cuda_visible_devices"] = "0,1"

2. 使用函式 set_device

##不建議使用該方法

import torch 
torch.cuda.set_device(1)

不過官方建議使用cuda_visible_devices，不建議使用 set_device 函式。

以上引用自：

2.如何將tensor或函式放在gpu上

直接使用tensor.cuda() 或 func().cuda()即可。不能直接對類或類物件執行".cuda()"

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...

Pytorch指定GPU的方法

export cuda visible devices 0 這裡是要使用的gpu編號2.在程式開頭設定 os.environ cuda visible devices 0,1,2,3 在執行程式時指定一般先在程式開頭輸入該 3.執行程式時使用命令列,來設定該程式可見的gpu cuda visible...