Closed JinguangHe closed 6 hours ago
You mean set your cuda device? You can set that with CUDA_VISIBLE_DEVICES . For torch device, like "cuda", "mps", "cpu", you can set via TORCH_DEVICE
You mean set your cuda device? You can set that with CUDA_VISIBLE_DEVICES . For torch device, like "cuda", "mps", "cpu", you can set via
TORCH_DEVICE
I have tried to modify settings.py
settings = Settings(CUDA_VISIBLE_DEVICES=1)
and created local.env to set "CUDA_VISIBLE_DEVICES=1"
But it didn't work, it always used gpu 0, not 1
You just export CUDA_VISIBLE_DEVICES=1 on the command line - this is an nvidia env variable, has nothing to do with this repo
Is there no parameter to assign TORCH_DEVICE 0 or 1 ?