下载地址：https://pytorch.org/get-started/locally/
PyTorch Stable 2.3.1
| 系统 | 平台 | 编程语言 | cuda版本 | 命令 |
|---|---|---|---|---|
| Linux | Conda | Python | 11.8 | conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia |
| Linux | Conda | Python | 12.1 | conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia |
| Linux | Conda | Python | ROCm 6.0 | NOTE: Conda packages are not currently available for ROCm, please use pip instead |
| Linux | Conda | Python | CPU | conda install pytorch torchvision torchaudio cpuonly -c pytorch |
| MacOS | Conda | Python | Default | conda install pytorch::pytorch torchvision torchaudio -c pytorch |
| Windows | Conda | Python | 11.8 | conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia |
| Windows | Conda | Python | 12.1 | conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia |
| Windows | Conda | Python | CPU | conda install pytorch torchvision torchaudio cpuonly -c pytorch |
| Linux | Pip | Python | 11.8 | pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 |
| Linux | Pip | Python | 12.1 | pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 |
| Linux | Pip | Python | ROCm 6.0 | pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0 |
| Linux | Pip | Python | CPU | pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu |
| Linux | LibTorch | C++/Java | 11.8 | Download here (Pre-cxx11 ABI): https://download.pytorch.org/libtorch/cu118/libtorch-shared-with-deps-2.3.1%2Bcu118.zip Download here (cxx11 ABI): https://download.pytorch.org/libtorch/cu118/libtorch-cxx11-abi-shared-with-deps-2.3.1%2Bcu118.zip |
| Linux | LibTorch | C++/Java | 12.1 | Download here (Pre-cxx11 ABI): https://download.pytorch.org/libtorch/cu121/libtorch-shared-with-deps-2.3.1%2Bcu121.zip Download here (cxx11 ABI): https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.3.1%2Bcu121.zip |
| Linux | LibTorch | C++/Java | ROCm 6.0 | Download here (Pre-cxx11 ABI): https://download.pytorch.org/libtorch/rocm6.0/libtorch-shared-with-deps-2.3.1%2Brocm6.0.zip Download here (cxx11 ABI): https://download.pytorch.org/libtorch/rocm6.0/libtorch-cxx11-abi-shared-with-deps-2.3.1%2Brocm6.0.zip |
| Linux | LibTorch | C++/Java | CPU | Download here (Pre-cxx11 ABI): https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-2.3.1%2Bcpu.zip Download here (cxx11 ABI): https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.3.1%2Bcpu.zip |
| Linux | Conda | Python |
Linux 平台
| 版本 | 平台 | 编程语言 | cuda版本 | 命令 |
|---|---|---|---|---|
| 2.3.1 | pip | python | 11.8 | pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 |
| 2.3.1 | conda | python | 11.8 | conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia |
| 2.3.1 | LibTorch | C++/Java | 11.8 | Download here (Pre-cxx11 ABI): https://download.pytorch.org/libtorch/cu118/libtorch-shared-with-deps-2.3.1%2Bcu118.zip Download here (cxx11 ABI): https://download.pytorch.org/libtorch/cu118/libtorch-cxx11-abi-shared-with-deps-2.3.1%2Bcu118.zip |
| 2.3.1 | Source | python | 11.8 | # Follow instructions at this URL: https://github.com/pytorch/pytorch#from-source |
| 2.3.1 | python | 11.8 | ||
| 2.3.1 | python | 11.8 | ||
| 2.3.1 | python | 11.8 |