RuntimeError: CUDA must not be initialized in the main process on Spaces with Stateless GPU environment.

#6
by mrfakename - opened
ZeroGPU Explorers org

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!

ZeroGPU Explorers org
This comment has been hidden
ZeroGPU Explorers org

OK actually it needs to be loaded on CPU and then moved to GPU

mrfakename changed discussion status to closed

Sign up or log in to comment