pytorch中設定使用指定的GPU

2022-11-24 15:31:34 字數 552 閱讀 6115

**:

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

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

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

1.1 直接終端中設定:

cuda_visible_devices=1 python main.py

1.2 python**中設定:

importos

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

見**:

2. 使用函式 set_device

import torch

torch.cuda.set_device(id)

該函式見 pytorch-master\torch\cuda\__init__.py。

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

gdb原始碼安裝,指定使用的python版本

gdb除錯python的時候,需要根據不同的python版本2.6 2.7 3.x安裝相應的gdb 如何指定關聯的python版本?下面gdb原始碼,解壓後,進入目錄 configure h 並沒有發現 with python的選項。沒有也沒有問題,沒有也可以自己加 which python找到py...