diff --git a/.github/workflows/dummy.yml b/.github/workflows/dummy.yml index 5ba5122c..af1ed757 100644 --- a/.github/workflows/dummy.yml +++ b/.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]