安装1.7 gpu版本的pytorch报错

安装1.7 gpu版本的pytorch报错 ERROR: torch has an invalid wheel, .dist-info directory not found
在这里插入图片描述
原因是cuda版本不一致,

解决方法1:

安装cpu版本的torch

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

解决方法2:

conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch

安装1.7 gpu版本的pytorch报错_第1张图片

你可能感兴趣的:(深度学习,日常bug系列,pytorch,深度学习)