【Python】CUDA11.6安装PyTorch三件套
由于PyTorch官网没有提供除适配CUDA10.3和11.3之外的安装方式,因此可以使用Nightly Binaries方式下载与自己CUDA版本合适的PyTorch
以CUDA11.6为例,在终端中输入以下指令:
1. 下载PyTorch
pip install torch --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116
2. 下载Torchvision
pip install torchvision --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116
3. 下载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
来源:ericdiii