Fix grid

Fixes an error that no CUDA devices are available.

In fact, CUDA might be available on the scheduler machine, and because this part of the code was run by the scheduler, torch.cuda.is_available() returned True, which caused the job to crash. This problem is solved by applying the modification in this merge request.

requested review from @amohammadi

assigned to @akomaty

added 1 commit

• 82793978 - fixed the device selection issue

Compare with previous version

changed the description

changed the description

merged

mentioned in commit 944573ab