Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions _additional_platforms/cuda_woa.json
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,11 @@
"name": "CUDA 13.4 ARM64",
"support_channel": "https://github.com/pytorch/pytorch/issues",
"stable": {
"windows": "python -m pip install \"torch==2.14.0+cu134\" \"torchvision==0.29.0+cu134\" \"torchaudio==2.11.0+cu134\" --extra-index-url https://pypi.nvidia.com/nvtorch_oot/"
"windows": "python -m pip install \"torch==2.15.0+cu134\" \"torchvision==0.29.0+cu134\" \"torchaudio==2.11.0+cu134\" --extra-index-url https://pypi.nvidia.com/nvtorch_oot/",
"linux": "Please go to getting started page for installation instructions: https://pytorch.org/get-started/locally/"
},
"preview": {
"windows": "python -m pip install --pre torch torchvision torchaudio --extra-index-url https://pypi.nvidia.com/nvtorch_oot_nightly/"
"windows": "python -m pip install --pre torch torchvision torchaudio --extra-index-url https://pypi.nvidia.com/nvtorch_oot_nightly/",
"linux": "Please go to getting started page for installation instructions: https://pytorch.org/get-started/locally/"
}
}
4 changes: 2 additions & 2 deletions _get_started/additional_platforms/cuda_woa.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
### pip

```bash
python -m pip install "torch==2.14.0+cu134" "torchvision==0.29.0+cu134" "torchaudio==2.11.0+cu134" --extra-index-url https://pypi.nvidia.com/nvtorch_oot/
python -m pip install "torch==2.15.0+cu134" "torchvision==0.29.0+cu134" "torchaudio==2.11.0+cu134" --extra-index-url https://pypi.nvidia.com/nvtorch_oot/
```

Use the `pip` package manager to install PyTorch, TorchVision and TorchAudio with CUDA support from the NVIDIA package index. Select your preferred options in the selector above to get the installation command. Nightly builds are available from `https://pypi.nvidia.com/nvtorch_oot_nightly/`.
Expand Down Expand Up @@ -50,7 +50,7 @@ else:
The following, or a similar output, indicates successful installation:

```bash
2.14.0+cu134
2.15.0+cu134
CUDA is available!
CUDA devices: 1
Device name: NVIDIA ...
Expand Down