From 59f9b6b850b532a88b829ceccda4c3e8aa228c71 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Fri, 2 Feb 2024 15:31:41 +0100 Subject: [PATCH] docker --- .github/workflows/dummy.yml | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/.github/workflows/dummy.yml b/.github/workflows/dummy.yml index 9f41adcd..65cdbc29 100644 --- a/.github/workflows/dummy.yml +++ b/.github/workflows/dummy.yml @@ -35,3 +35,18 @@ jobs: - name: show directory run: | ls -l /mnt/cache/ + + github_repo: + name: github repo + 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: clone + run: | + pwd + git clone https://github.com/huggingface/Google-Cloud-Containers.git + cd Google-Cloud-Containers + ls -la