安装1.7 gpu版本的pytorch报错_error: torch has an invalid wheel, .dist-info dire

作者：你好赵伟 | 2024-05-17 23:58:47

踩

error: torch has an invalid wheel, .dist-info directory not found

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

解决方法2：

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

在这里插入图片描述

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/你好赵伟/article/detail/585744