Improve compatibility with CUDA-enabled pytorch on non-CUDA devices about cleanrl HOT 3 CLOSED

Could you clarify why this is a "race condition"?

it shouldn't be cuda unless both cases are true so I don't understand why ordering would matter in this case

from cleanrl.

Because Python's and is lazy. If the first option is false, than it doesn't evaluate the second one. If I pass --no-cuda, Python still runs torch.cuda.is_available, even though I specified that I do not want to use CUDA.

On devices that don't have CUDA drivers (or have the stub drivers) but have the CUDA version of PyTorch installed, this throws a runtime error. However, using the CPU on such devices is valid, since the PyTorch library can still function by using the CPU.

from cleanrl.

Ok, that makes sense, I thought that pytorch would be smart enough to not raise a runtime error for this function.
I would make a PR that makes your suggested change

from cleanrl.