ImportError: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory

心中带点小风骚 • 2023年11月6日上午10:42 • Python • 阅读 158

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

mmcv-full install doc

文章出处登录后可见！

已经登录？立即刷新

上一篇 2023年11月6日

下一篇 2023年11月6日