This commit is contained in:
AbdBarho 2022-10-14 16:29:46 +02:00
parent fca33d5c04
commit 79c79f87cb

View file

@ -23,7 +23,7 @@ jobs:
wget https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.0-1_all.deb 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 dpkg -i cuda-keyring_1.0-1_all.deb
apt-get update 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 export PIP_CACHE_DIR=$(pwd)/cache
pip install ninja install torch --extra-index-url https://download.pytorch.org/whl/cu113 pip install ninja install torch --extra-index-url https://download.pytorch.org/whl/cu113