We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 169aa0b commit 252dafcCopy full SHA for 252dafc
dockerized-llama-tornado-nvidia-opencl
@@ -2,7 +2,7 @@
2
3
IMAGE=beehivelab/gpullama3.java-nvidia-openjdk-opencl:latest
4
if [[ "$1" == "--console" ]]; then
5
- exec docker run --runtime=nvidia --rm -it -v "$PWD":/data "$IMAGE"
+ exec docker run --gpus all --rm -it -v "$PWD":/data "$IMAGE"
6
else
7
- exec docker run --runtime=nvidia --rm -it -v "$PWD":/data "$IMAGE" "/gpullama3/GPULlama3.java/llama-tornado" "$@"
+ exec docker run --gpus all --rm -it -v "$PWD":/data "$IMAGE" "/gpullama3/GPULlama3.java/llama-tornado" "$@"
8
fi
0 commit comments