question about CUDA unavailable

[Z0Victor]

when I run Main.py , issue comes up :

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.

how to deal with that??PLZ

[AroundKiki]

1. try torch.cuda.is_available() in your python environment, it should be True.
2. in Main.py, the default "device" params is "cuda:1", you should change it to "cuda" or "cuda:0" if you have single GPU installed.