去官网下载:链接
根据CUDA的版本下载对应的pytorch和torchvision
# CUDA 9.2
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=9.2 -c pytorch
# CUDA 10.1
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
# CUDA 10.2
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.2 -c pytorch
# CUDA 11.0
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
# CPU Only
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cpuonly -c pytorch
直接终端下载很慢,建议复制链接去window下载,再回到ubuntu安装
本文提供了根据不同CUDA版本安装PyTorch、torchvision和torchaudio的具体命令。包括CUDA9.2、10.1、10.2、11.0及仅CPU版本的安装指导。
1万+

被折叠的 条评论
为什么被折叠?



