from pytorch3D import ***报错 ImportError: libtorch_cuda_cu.so: cannot open shared object file

问题分析

在编写代码时，我使用这个命令后“from pytorch3d.ops import knn_gather”，报了下方错误“ImportError: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory”，查阅了相关资料后，可能的问题是pytorch3D版本不对，或者与cuda版本不匹配。

解决流程

打开终端，进入自己使用的环境，键入下方命令分析自己电脑中的cuda和pytorch版本

python -c "import torch; import pytorch3d; print('PyTorch version:', torch.__version__)"

我的终端中打印出下方信息，pytorch版本为201，cuda版本为11.7

PyTorch version: 2.0.1+cu117

此时还需要明确自己的python版本，我就不列具体方法了

接下来去pytorch3D官网寻找匹配版本

https://anaconda.org/pytorch3d/pytorch3d/files?page=2

此时需要注意你选择的版本必须满足3个要求

1.python版本要一致

2.cuda版本要一致

3.pytorch版本要一致

确定好版本后，鼠标右击复制链接地址

然后到终端框中（请注意环境必须是报错的那个），然后在下方命令之后添加刚才复制的链接即可

conda install 

我使用的代码如下

conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.8/download/linux-64/pytorch3d-0.7.8-py312_cu121_pyt222.tar.bz2

键入下方代码可查看pytorch3D版本信息

python -c "import torch; import pytorch3d; print('PyTorch3D version:', pytorch3d.__version__)"