PyTorch 2.2 RC1 is produced for pytorch, audio, vision, and text

They are now available for download from pytorch-test channel.

Reminder of key dates:

  • M3.3: Domain libraries cut RC Branch (Dec 08th 2023) - COMPLETED
  • M4: Release branch finalized, Announce final launch date, Feature classifications published (week of Dec 18th 2023)
  • M5: External-Facing Content Finalized (Jan 8th 2024)
  • M6: Release Day (Jan 11th 2024)

CONDA CPU

Windows/Linux:
conda install pytorch=2.2.0 torchvision torchaudio cpuonly -c pytorch-test

MacOS:
conda install pytorch=2.2.0 torchvision torchaudio -c pytorch-test

CUDA 11.8, 12.1

Windows/Linux:
conda install pytorch=2.2.0 torchvision torchaudio pytorch-cuda=11.8 -c pytorch-test -c nvidia
conda install pytorch=2.2.0 torchvision torchaudio pytorch-cuda=12.1 -c pytorch-test -c nvidia

PIP CPU

Windows/Linux/MacOS:
pip3 install torch==2.2.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cpu

PIP CUDA 11.8, 12.1

Windows/Linux:
pip3 install torch==2.2.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cu118
pip3 install torch==2.2.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cu121

PIP ROCM 5.6
pip3 install torch==2.2.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/test/rocm5.6

PIP Linux Aarch64
pip3 install torch==2.2.0 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/test/cpu

Libtorch

CPU Linux:

(Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/test/cpu/libtorch-shared-with-deps-latest.zip

(cxx11 ABI):
https://download.pytorch.org/libtorch/test/cpu/libtorch-cxx11-abi-shared-with-deps-latest.zip

CPU Windows:

Download here (Release version):
https://download.pytorch.org/libtorch/test/cpu/libtorch-win-shared-with-deps-latest.zip

Download here (Debug version):
https://download.pytorch.org/libtorch/test/cpu/libtorch-win-shared-with-deps-debug-latest.zip

CPU MacOS:
https://download.pytorch.org/libtorch/test/cpu/libtorch-macos-latest.zip

CUDA 11.8, 12.1 Linux

(Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/test/cu118/libtorch-shared-with-deps-latest.zip
https://download.pytorch.org/libtorch/test/cu121/libtorch-shared-with-deps-latest.zip

(cxx11 ABI):
https://download.pytorch.org/libtorch/test/cu118/libtorch-cxx11-abi-shared-with-deps-latest.zip
https://download.pytorch.org/libtorch/test/cu121/libtorch-cxx11-abi-shared-with-deps-latest.zip

Windows CUDA 11.8, 12.1

(Release version):
https://download.pytorch.org/libtorch/test/cu118/libtorch-win-shared-with-deps-latest.zip
https://download.pytorch.org/libtorch/test/cu118/libtorch-win-shared-with-deps-latest.zip

(Debug version):
https://download.pytorch.org/libtorch/test/cu118/libtorch-win-shared-with-deps-debug-latest.zip
https://download.pytorch.org/libtorch/test/cu121/libtorch-win-shared-with-deps-debug-latest.zip

If you have any questions please comment below or reach out to us on this thread.

1 Like