Skip to content

Commit

Permalink
Fix cuda not being used on windows: update pytorch version (#6064)
Browse files Browse the repository at this point in the history
These references were missed when upgrading from pytorch 1.x to 2.x in #6013

References found by running `grep -R '1\.13\.1' .`

Install command chosen from the guide at https://pytorch.org/get-started/locally/
  • Loading branch information
alex-mccarthy-unity authored Mar 6, 2024
1 parent 6004556 commit 1bee58f
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 3 deletions.
2 changes: 1 addition & 1 deletion .yamato/pytest-gpu.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117 --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
if python -c "exec('import torch \nif not torch.cuda.is_available(): raise')" &> /dev/null; then
echo 'all good'
else
Expand Down
6 changes: 4 additions & 2 deletions docs/Installation.md
Original file line number Diff line number Diff line change
Expand Up @@ -146,10 +146,12 @@ offer a dedicated [guide on Virtual Environments](Using-Virtual-Environment.md).
#### (Windows) Installing PyTorch

On Windows, you'll have to install the PyTorch package separately prior to
installing ML-Agents. Activate your virtual environment and run from the command line:
installing ML-Agents in order to make sure the cuda-enabled version is used,
rather than the CPU-only version. Activate your virtual environment and run from
the command line:

```sh
pip3 install torch~=1.13.1 -f https://download.pytorch.org/whl/torch_stable.html
pip3 install torch~=2.2.1 --index-url https://download.pytorch.org/whl/cu121
```

Note that on Windows, you may also need Microsoft's
Expand Down

0 comments on commit 1bee58f

Please sign in to comment.