Expected all tensors to be on the same device, but found at least two devices.

[ChiariRay]

Describe the bug Failed to run inference in MONAI Label Server, reported RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument index in method wrapper_CUDA__index_select).

To Reproduce According to the tutorial in README. Steps to reproduce the behavior:

1. Start the server: monailabel start_server --app monaivista --studies ./Task07_Pancreas/imagesTs --conf models vista_point_2pt5
2. Connect to the monailabel server in 3D slicer, and click Next Sample
3. Click Run

Screenshots

Additional context Run the server on ubuntu 22.04 with 2 RTX 4090 Run the 3D slicer on MacOS 14.0