GPU 리소스를 특정 프로세스로 제한

GPU 리소스를 특정 프로세스로 제한

ulimit명령은 Linux 시스템의 프로세스 메모리를 제한할 수 있습니다. 마찬가지로 Linux에는 제한된 GPU 리소스로 프로세스를 실행하는 명령이 있습니까? 예를 들어, 64GB GPU 시스템이 있는데 몇 가지 실험적인 이유로 특정 프로세스(YOLO)가 1GB GPU만 사용하도록 제한하고 싶습니다.

관련 정보