ubuntu+anaconda+pytorch+cuda找不到libcupti.so.12、libcupti.so.2等

的nvidia文件夹复制到/home/name/anaconda3/envs/pytorch/lib/python3.10/site-packages目录下。则将/home/name/.local/lib/python3.10/site-packages目录。如创建pytorch环境使用python3.10。

qq_52487425

191人浏览 · 2024-11-03 20:32:51

qq_52487425 · 2024-11-03 20:32:51 发布

报错

from torch._C import * # noqa: F403；
ImportError: libcupti.so.12:cannot open shared object file: No such file or directory
ImportError: libcupti.so.2:cannot open shared object file: No such file or directory

解决方法：
如创建pytorch环境使用python3.10
则将/home/name/.local/lib/python3.10/site-packages目录
的nvidia文件夹复制到/home/name/anaconda3/envs/pytorch/lib/python3.10/site-packages目录下

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')..