From e39d20ab6c21df509ab31fabd4ab0498589e4356 Mon Sep 17 00:00:00 2001 From: alex-mccarthy-unity Date: Wed, 6 Mar 2024 16:42:57 +0100 Subject: [PATCH] Fix GPU continuous build: correct torchaudio version torchaudio 0.17 doesn't exist, but 2.2 does. (I'm not sure if the tests actually need torchaudio or torchvision, since those aren't listed dependencies of any of our software, but let's at least install a valid one) --- .yamato/pytest-gpu.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.yamato/pytest-gpu.yml b/.yamato/pytest-gpu.yml index 2ab36e7182..d3a5474743 100644 --- a/.yamato/pytest-gpu.yml +++ b/.yamato/pytest-gpu.yml @@ -11,7 +11,7 @@ pytest_gpu: python3 -m pip install pyyaml --index-url https://artifactory.prd.it.unity3d.com/artifactory/api/pypi/pypi/simple python3 -u -m ml-agents.tests.yamato.setup_venv python3 -m pip install --progress-bar=off -r test_requirements.txt --index-url https://artifactory.prd.it.unity3d.com/artifactory/api/pypi/pypi/simple - python3 -m pip install torch==2.2.1+cu121 torchvision==0.17.1+cu121 torchaudio==0.17.1 --index-url https://download.pytorch.org/whl/cu121 + python3 -m pip install torch==2.2.1+cu121 torchvision==0.17.1+cu121 torchaudio==2.2.1+cu121 --index-url https://download.pytorch.org/whl/cu121 if python -c "exec('import torch \nif not torch.cuda.is_available(): raise')" &> /dev/null; then echo 'all good' else