Closed Aksh97 closed 2 years ago
Hi, which parameter needs to be passed to enable GPU for inference?
CUDA_VISIBLE_DEVICES=0 python xx.py
This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs.
Hi, which parameter needs to be passed to enable GPU for inference?