# CUDA 10.1
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
cuda10.1对应pytorch版本1.8.1,torchvision为0.9.1
清华源下载地址
https://pypi.tuna.tsinghua.edu.cn/simple
# CUDA 10.1
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
cuda10.1对应pytorch版本1.8.1,torchvision为0.9.1
清华源下载地址
https://pypi.tuna.tsinghua.edu.cn/simple