Closed kwea123 closed 5 years ago
add --rm tag to automatically remove the container after the scripts terminate, and add --runtime=nvidia to enable GPU when launching docker
--rm
--runtime=nvidia
add
--rm
tag to automatically remove the container after the scripts terminate, and add--runtime=nvidia
to enable GPU when launching docker