PyTorch指定GPU裝置

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

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

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

1.1 直接終端中設定：

cuda_visible_devices=1 python main.py

import os

os.environ["cuda_visible_devices"] = "2"

2. 使用函式 set_device

import torch

torch.cuda.set_device(id)

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

使用指定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

pytorch預設使用從0開始的gpu，如果gpu0正在執行程式，需要指定其他gpu。有如下兩種方法來指定需要使用的gpu。1.類似tensorflow指定gpu的方式，使用cuda visible devices。1.1 直接終端中設定 cuda visible devices 1 python ...

Pytorch指定GPU的方法

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