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 252dafc commit 1ca6901Copy full SHA for 1ca6901
dockerized-llama-tornado-nvidia-ptx
@@ -2,7 +2,7 @@
2
3
IMAGE=beehivelab/gpullama3.java-nvidia-openjdk-ptx: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