由于PyTorch 官网没有提供除适配CUDA10.3和11.3之外的安装方式,因此可以使用 Nightly Binaries方式下载与自己CUDA版本合适的PyTorch
以CUDA11.6为例,在终端中输入以下指令:
- 下载PyTorch
pip install torch --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116
- 下载Torchvision
pip install torchvision --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116
- 下载Torchaudio
pip install torchaudio --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116
或直接下载三件套:
pip install torch torchvision torchaudio --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116
检查PyTorch是否安装成功:
参考:
[1] PyTorch + CUDA 11.6 – deployment – PyTorch Forums
Original: https://blog.csdn.net/ericdiii/article/details/125258580
Author: ericdiii
Title: 【Python】CUDA11.6安装PyTorch三件套
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/668490/
转载文章受原作者版权保护。转载请注明原作者出处!