diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml index e6edcbe..8d0dfca 100644 --- a/.github/workflows/docker.yml +++ b/.github/workflows/docker.yml @@ -23,7 +23,7 @@ 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-3 cuda-libraries-dev-11-3 -y + apt-get install cuda-nvcc-11-8 cuda-libraries-dev-11-8 -y export PIP_CACHE_DIR=$(pwd)/cache pip install ninja install torch --extra-index-url https://download.pytorch.org/whl/cu113