Skip to content

Commit 1ca6901

Browse files
authored
[ptx] Update runtime flag in favor of modern --gpus all
1 parent 252dafc commit 1ca6901

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dockerized-llama-tornado-nvidia-ptx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
IMAGE=beehivelab/gpullama3.java-nvidia-openjdk-ptx:latest
44
if [[ "$1" == "--console" ]]; then
5-
exec docker run --runtime=nvidia --rm -it -v "$PWD":/data "$IMAGE"
5+
exec docker run --gpus all --rm -it -v "$PWD":/data "$IMAGE"
66
else
7-
exec docker run --runtime=nvidia --rm -it -v "$PWD":/data "$IMAGE" "/gpullama3/GPULlama3.java/llama-tornado" "$@"
7+
exec docker run --gpus all --rm -it -v "$PWD":/data "$IMAGE" "/gpullama3/GPULlama3.java/llama-tornado" "$@"
88
fi

0 commit comments

Comments
 (0)