docker

.github/workflows/dummy.yml

@@ -42,6 +42,23 @@ jobs:
 #        run: |
 #          ls -l /mnt/cache/
 
+  container_gcp:
+    name: load docker image
+    runs-on: [single-gpu, nvidia-gpu, a10, ci]
+    container:
+      # TODO: make this $ {{ inputs.image }}
+      image: us-central1-docker.pkg.dev/gcp-partnership-412108/deep-learning-images/huggingface-pytorch-training-gpu.2.1.transformers.4.37.2.py310:latest
+      # options: --gpus all --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
+    steps:
+      - name: NVIDIA-SMI
+        run: |
+          echo "Hello A10"
+          nvidia-smi
+
+      - name: show directory
+        run: |
+          ls -l ./
+
   github_repo:
     name: github repo
     runs-on: [single-gpu, nvidia-gpu, a10, ci]