ipdate

hisashi-ito · Mar 20, 2023 · 8b46d47 · 8b46d47
1 parent 376e128
commit 8b46d47
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/build.sh b/build.sh
@@ -1,3 +1,3 @@
 #! /bin/bash
-# sudo docker build --no-cache -t gpt-neox .
-sudo docker build -t gpt-neox .
+sudo docker build --no-cache -t gpt-neox .
+#sudo docker build -t gpt-neox .
diff --git a/launch.sh b/launch.sh
@@ -3,11 +3,12 @@ IMAGE="gpt-neox"
 CONTAINER="gpt-neox"
 sudo docker run -tid \
      --privileged \
+     --gpus all \
      -v /data:/data \
+     --shm-size=64gb \
      --network=host \
      --name ${CONTAINER} \
      ${IMAGE} /bin/bash
 
 # -p 8888:8888 \
 # --shm-size=64gb \
-# --network=host \