Skip to content

Commit 5366a98

Browse files
committed
use cuda:12.1.1-cudnn8-devel-ubuntu22.04 and fix home dir in run.sh
1 parent f1b9db7 commit 5366a98

File tree

2 files changed

+4
-7
lines changed

2 files changed

+4
-7
lines changed

docker/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu20.04
1+
FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
22

33
ARG DEBIAN_FRONTEND=noninteractive
44
ENV TZ=Asia/Taipei

docker/run.sh

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,9 @@ if [ "$(docker ps -a -q -f name=${CONTAINER_NAME})" ]; then
1515
#docker exec -it ${CONTAINER_NAME} "$@"
1616
else
1717
echo "Creating and running a new container ${CONTAINER_NAME}..."
18-
# mount the current directory to /workspace to prevent some permission issues
19-
# echo "mount ${PWD} to /workspace"
2018
docker run -it --gpus all --name ${CONTAINER_NAME} \
21-
--user $(id -u):$(id -g)
2219
-v "${PWD}:/home/user/Quamba:rw" \
20+
-w /home/user/Quamba \
2321
--shm-size 64G \
24-
"${IMAGE_NAME}" \
25-
bash -c "cd /workspace && exec bash"
26-
fi
22+
"${IMAGE_NAME}" $@
23+
fi

0 commit comments

Comments
 (0)