How to set gs.device?

[HerocatUED]

I use 8 Nvidia-4090 GPU, but I need to set gs.device to other GPU(e.g. GPU5) because my GPU0 is bussy. I can set all the tensors to device 5, but anything I got from gs are on device 0. So how can I set gs.device to other devices rather than device0? I didn't find any tutorial on doc...

[Kashu7100]

You can set environment variables using os.environ. Do the following before initializing.

import os
# os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="5"

[HerocatUED]

Thanks!