try

spaces/zero/torch.py

@@ -241,14 +241,14 @@ if (torch := maybe_import_torch()):
         bitsandbytes.unpatch()
 
     def _move(nvidia_uuid: str):
-        os.environ['CUDA_VISIBLE_DEVICES'] = "0,1,2,3,4,5"
+        os.environ['CUDA_VISIBLE_DEVICES'] = "5,8,10,11,14,15"
         import subprocess
         subprocess.run(
             ['nvidia-smi'], text=True
         )
 
         print("_move(nvidia_uuid: str)")
-        torch.Tensor([0]).cuda() # CUDA init
+        torch.Tensor([5]).cuda() # CUDA init
         for op in to_ops.items():
             tensor, parsed_args = op
             _, dtype, _, memory_format = parsed_args