From fca33d5c0455171c20cf76628f28d2759439e290 Mon Sep 17 00:00:00 2001 From: AbdBarho Date: Fri, 14 Oct 2022 16:28:57 +0200 Subject: [PATCH] i --- .github/workflows/docker.yml | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml index 8a7de5e..e6edcbe 100644 --- a/.github/workflows/docker.yml +++ b/.github/workflows/docker.yml @@ -23,10 +23,11 @@ jobs: wget https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.0-1_all.deb dpkg -i cuda-keyring_1.0-1_all.deb apt-get update - apt-get install cuda-nvcc-11-8 cuda-libraries-dev-11-8 -y + apt-get install cuda-nvcc-11-3 cuda-libraries-dev-11-3 -y + + export PIP_CACHE_DIR=$(pwd)/cache + pip install ninja install torch --extra-index-url https://download.pytorch.org/whl/cu113 - export PIP_CACHE_DIR=$(pwd) - pip install ninja torch==1.12.1+cu113 pip wheel --wheel-dir=data git+https://github.com/facebookresearch/xformers.git@ba93c5012d00bd1b010514a7bc9bd938c1ad6149#egg=xformers - name: Archive production artifacts uses: actions/upload-artifact@v3