Skip to content

Commit 8e0108a

Browse files
author
Ervin T
authored
[docs] Update install docs for PyTorch (#4488)
1 parent 4f201b0 commit 8e0108a

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

docs/Training-ML-Agents.md

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -565,15 +565,15 @@ be installed, and PyTorch must be enabled for your trainer.
565565

566566
#### Installing PyTorch
567567

568-
If you've already installed ML-Agents, installing PyTorch can be
569-
done by running `pip3 install torch` after activating your virtual
570-
environment.
571-
572-
If you're installing or upgrading ML-Agents, run `pip3 install mlagents[torch]`
573-
instead of `pip3 install mlagents` during [installation](Installation.md).
574-
575-
We support versions 1.6.0 and later.
576-
See the [official install instructions](https://pytorch.org/) for more info.
568+
If you've already installed ML-Agents, follow the
569+
[official PyTorch install instructions](https://pytorch.org/get-started/locally/) for
570+
your platform and configuration. Note that on Windows, you may also need Microsoft's
571+
[Visual C++ Redistributable](https://support.microsoft.com/en-us/help/2977003/the-latest-supported-visual-c-downloads) if you don't have it already.
572+
573+
If you're installing or upgrading ML-Agents on Linux or Mac, you can also run
574+
`pip3 install mlagents[torch]` instead of `pip3 install mlagents`
575+
during [installation](Installation.md). On Windows, install ML-Agents first and then
576+
separately install PyTorch.
577577

578578
#### Enabling PyTorch
579579

0 commit comments

Comments
 (0)