ImportError: libtorch

• Python

原因

cuda版本选的不对

解决

python -c 'import torch;print(torch.__version__);print(torch.version.cuda)'

查看cuda版本和torch版本

我的输出如下：

1.9.0+cu102
10.2

用pip安装时按照上面的选就行

pip install mmcv-full

ref

mmcv-full install doc

本文来自网络，不代表协通编程立场，如若转载，请注明出处：https://net2asp.com/ff035e6940.html

赞 (0)
« 上一篇 1天前
下一篇 » 1天前

相关推荐