docker

.github/workflows/dummy.yml

@@ -6,11 +6,32 @@ on:
       - dummy_branch
 
 jobs:
-  setup:
-    name: check
+  echo:
+    name: show host
     runs-on: [single-gpu, nvidia-gpu, a10, ci]
     steps:
       - name: NVIDIA-SMI
         run: |
           echo "Hello A10"
           nvidia-smi
+
+      - name: show directory
+        run: |
+          ls -l /mnt/cache/.cache/
+
+  container:
+    name: load docker image
+    runs-on: [single-gpu, nvidia-gpu, a10, ci]
+    container:
+      # TODO: make this $ {{ inputs.image }}
+      image: huggingface/transformers-all-latest-gpu
+      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 /mnt/cache/