安装cuda10.1对应的pytorch

1.直接pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio===0.7.0报错如下:

解决方案如下: 

pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

你可能感兴趣的:(pytorch,深度学习,python)