GPUリソ​​ースを特定のプロセスに制限する

このulimitコマンドは、Linuxシステムのプロセスメモリを制限できます。同様に、Linuxには限られたGPUリソ​​ースでプロセスを実行するコマンドがありますか？たとえば、64GB GPUシステムがありますが、いくつかの実験的な理由から、特定のプロセス（YOLO）が1GB GPUのみを使用するように制限したいと思います。