官网安装指南：https://pytorch.org/get-started/locally/
1. Linux
| Language | Package | CUDA | Command |
|---|---|---|---|
| Python | pip | 9.2 | pip install torch==1.7.1+cu92 torchvision==0.8.2+cu92 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html |
| 10.1 | pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html | ||
| 10.2 | |||
| 11.0 | pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html | ||
| None | pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html | ||
| conda | 9.2 | conda install pytorch torchvision torchaudio cudatoolkit=9.2 -c pytorch | |
| 10.1 | conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch | ||
| 10.2 | conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch | ||
| 11.0 | conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch | ||
| None | conda install pytorch torchvision torchaudio cpuonly -c pytorch |
1.2 Windows
版权声明
本站仅做备份收录，仅供研究与教学参考之用。
读者将信息用于其他用途的，全部法律及连带责任由读者自行承担，本站不承担任何责任。
评论