安装AimetTorch报错:No matching distribution found for torch==1.9.0+cu111

问题:安装AimetTorch报错No matching distribution found for torch==1.9.0+cu111(torch和cuda版本均匹配)

原因:在轮子限制了具体版本

方法:打开AimetTorch-torch_gpu_1.22.1-cp38-cp38-linux_x86_64.whl/AimetTorch-torch_gpu_1.22.1.dist-info/METADATA

Requires-Dist: torch (==1.9.1+cu111)
Requires-Dist: torchvision (==0.10.1+cu111)

改成

Requires-Dist: torch (==1.9.1)
Requires-Dist: torchvision (==0.10.1)

你可能感兴趣的:(深度学习,pytorch,python)