【配环境】pytorch3d版本不匹配导致ImportError: libtorch_cuda_cu.so: cannot open shared object file

之前直接用pip install下载的pytorch3d版本和环境不匹配，需要考虑到cuda、pytorch、python版本。

果砸 the alchemist

3184人浏览 · 2024-01-11 12:05:52

果砸 the alchemist · 2024-01-11 12:05:52 发布

报错

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

原因

之前直接用pip install下载的pytorch3d版本和环境不匹配，需要考虑到cuda、pytorch、python版本。

解决方式

最稳妥的方式就是直接官网下载。

到Anaconda官网https://anaconda.org/pytorch3d/pytorch3d/files，下载版本匹配的pytorch3d文件，右键复制链接
替换成复制的链接即可

conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.5/download/linux-64/pytorch3d-0.7.5-py38_cu118_pyt201.tar.bz2

Triton中文社区

欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。

更多推荐

torch.device(‘cuda:0‘ if cuda else ‘cpu‘)

# 这个地方的cuda:0 实际上并不是0号GPU，他取决于CUDA_VISIBLE_DEVICES# 然后逻辑GPU和物理GPU有一个对应关系# 如果CUDA_VISIBLE_DEVICES为2,1,3# 那么CUDA:0就是2号GPU， CUDA:1 就是1号GPU CUDA:3 就是3号GPUreturn torch.device('cuda:0' if cuda else 'cpu')..