Skip to content

Commit 548c15f

Browse files
committed
Add runner scripts for using containers
1 parent cfe0097 commit 548c15f

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed
Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
#!/usr/bin/env bash
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"
6+
else
7+
exec docker run --runtime=nvidia --rm -it -v "$PWD":/data "$IMAGE" "/gpullama3/GPULlama3.java/llama-tornado" "$@"
8+
fi
Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
#!/usr/bin/env bash
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"
6+
else
7+
exec docker run --runtime=nvidia --rm -it -v "$PWD":/data "$IMAGE" "/gpullama3/GPULlama3.java/llama-tornado" "$@"
8+
fi

0 commit comments

Comments
 (0)