diff --git a/run.sh b/run.sh index 8db2f9b..d81da0d 100755 --- a/run.sh +++ b/run.sh @@ -1,3 +1,3 @@ #!/bin/bash -docker run --device=/dev/kfd --device=/dev/dri --privileged --security-opt seccomp=unconfined --group-add video --shm-size 20g --rm -it --name axolotl --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --mount type=bind,src="/home/sb/lm/",target=/workspace/data --mount type=bind,src="/home/sb/svn/lm-finetuner/",target=/workspace/lm-finetuner pytorch-rocm-exo:v1 +docker run --device=/dev/kfd --device=/dev/dri --privileged --security-opt seccomp=unconfined --group-add video --shm-size 20g --rm -it --name lmcontainer --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --mount type=bind,src="/home/sb/lm/",target=/workspace/data --mount type=bind,src="/home/sb/svn/lm-finetuner/",target=/workspace/lm-finetuner pytorch-rocm-exo:v1