安装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