Spaces:
Running
Running
RuntimeError: CUDA must not be initialized in the main process on Spaces with Stateless GPU environment.
#6
by
mrfakename
- opened
Hi,
I duplicated the Pheme space and made the modifications for ZERO (import spaces
, @spaces.GPU(enable_queue=True
), however I got an error RuntimeError: CUDA must not be initialized in the main process on Spaces with Stateless GPU environment.
. Is there any way to get around this?
Thanks!
This comment has been hidden
OK actually it needs to be loaded on CPU and then moved to GPU
mrfakename
changed discussion status to
closed