Commit 92cbaad

authored

Fix GPU continuous build: correct torch version, and remove torchaudio and torchvideo (#6074)

* Fix GPU continuous build: correct torchaudio version torchaudio 0.17 doesn't exist, but 2.2 does. Use a slightly older cuda version, since that makes cuda detection work on the machines we're running on (RTX 2080's on Ubuntu 18.04, which presumably aren't compatible with CUDA 12). (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) Passing yamato run with this change: https://unity-ci.cds.internal.unity3d.com/job/34869354/logs * Don't install torchaudio and torchvision, since they're unused Sample passing GPU test run: https://unity-ci.cds.internal.unity3d.com/job/34891013/logs

1 parent b9ac3f1 commit 92cbaadCopy full SHA for 92cbaad

File tree

1 file changed

-1

lines changed

.yamato
- pytest-gpu.yml

1 file changed

-1

lines changed

`‎.yamato/pytest-gpu.yml‎`

Lines changed: 2 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -11,10 +11,11 @@ pytest_gpu:`
`11`	`11`	`python3 -m pip install pyyaml --index-url https://artifactory.prd.it.unity3d.com/artifactory/api/pypi/pypi/simple`
`12`	`12`	`python3 -u -m ml-agents.tests.yamato.setup_venv`
`13`	`13`	`python3 -m pip install --progress-bar=off -r test_requirements.txt --index-url https://artifactory.prd.it.unity3d.com/artifactory/api/pypi/pypi/simple`
`14`		`- 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`
	`14`	`+ python3 -m pip install torch==2.2.1+cu118 --index-url https://download.pytorch.org/whl/cu118`
`15`	`15`	`if python -c "exec('import torch \nif not torch.cuda.is_available(): raise')" &> /dev/null; then`
`16`	`16`	`echo 'all good'`
`17`	`17`	`else`
	`18`	`+ echo 'cuda device not available!'`
`18`	`19`	`exit 1`
`19`	`20`	`fi`
`20`	`21`	`python3 -m pytest -m "not slow" --junitxml=junit/test-results.xml -p no:warnings`

0 commit comments

Comments

(0)

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commit 92cbaad

File tree

1 file changed

1 file changed

`‎.yamato/pytest-gpu.yml‎`

0 commit comments