Skip to content

Commit 4a32bbc

Browse files
authored
Use nvidia as default runtime in the run-manager (PsycheFoundation#577)
1 parent 166563d commit 4a32bbc

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tools/rust-tools/run-manager/src/docker/manager.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,7 @@ impl RunManager {
153153
.arg("--network=host")
154154
.arg("--shm-size=1g")
155155
.arg("--privileged")
156+
.arg("--runtime=nvidia")
156157
.arg("--gpus=all")
157158
.arg("--device=/dev/infiniband:/dev/infiniband")
158159
.arg("--env")

0 commit comments

Comments
 (0)