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

原因

cuda版本选的不对

解决

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

查看cuda版本和torch版本
我的输出如下:

1.9.0+cu102
10.2

用pip安装时按照上面的选就行
ImportError: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory_第1张图片

ref

mmcv-full install doc

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