Skip to content

Commit 252dafc

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

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dockerized-llama-tornado-nvidia-opencl

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-opencl: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)