Replies: 1 comment 2 replies
-
CUDA_VISIBLE_DEVICES=0,2,3 python tools/train.py xxxx.py |
Beta Was this translation helpful? Give feedback.
2 replies
-
|
Where do I set this in the train.py? Is there an option to set it in the config file? |
Beta Was this translation helpful? Give feedback.
-
|
only can set in the environment values ， such as CUDA_VISIBLE_DEVICES=0,2,3 python tools/train.py xxxx.py
# or
export CUDA_VISIBLE_DEVICES=0,2,3
|
Beta Was this translation helpful? Give feedback.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Can mmyolo specify the gpu?
mmyolo 能指定gpu吗？
Beta Was this translation helpful? Give feedback.
All reactions