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