🧾 에러 내용

가. 해당 코드

if device >= 0:
        torch._C._cuda_setDevice(device)

나. Error Log

RuntimeError: CUDA error: invalid device ordinal

CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1
Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.

🤔 원인

ordinal : 서수 ( 첫째, 둘째 이런 것)

적절하지 않은 device 서수란다. device를 출력해봤을 때, 1이 나오는데 _cuda_setDevice(1)이 실행이 안되는것 같다.

✅ 해결 방법

사용가능한 GPU의 개수가 1개인데

torch._C._cuda_setDevice(device)

여기서 device가 1이라서 에러가 떴다.

device = _get_device_index(device)

아마도 위에 코드에서 인덱스를 1을 반환했나보다.

따라서 아래와 같이 수정해주었다.

torch._C._cuda_setDevice(device-1 or 0)