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

在这里插入图片描述

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐