Torch XLA CUDA

Creator

Creator

Seonglae Cho

Created

Created

2024 Feb 7 9:48

Editor

Editor

Seonglae Cho

Edited

Edited

2024 Feb 27 14:51

Refs

Refs

CUDA 1 1


export GPU_NUM_DEVICES=1 PJRT_DEVICE=GPU

CUDA 1 2


export GPU_NUM_DEVICES=1 PJRT_DEVICE=CUDA

Latest

https://github.com/pytorch/xla/blob/main/docs/gpu.md

CUDA 11

https://github.com/pytorch/xla/blob/2c11351beaffef52217c956e1ded0b2eb642dd2d/docs/gpu.md

Recommendations

////////