编译自定义算子出现的问题(GCFSR)-- libcudart.so.12: No such file

一、问题描述

  File "/home/jankin/miniconda3/envs/env_GCFSR/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1929, in _import_module_from_library
    module = importlib.util.module_from_spec(spec)
    
ImportError: libcudart.so.12: cannot open shared object file: No such file or directory

二、问题分析

1、调用的package 与 cuda 版本不匹配

本文中是在torch调用中报错,在torch官网中发现,在本博客创建时间时,pytorch支持的最高版本的cuda是11.8,对cuda-12是不支持的。

2、在进行与cuda相关的编译时,调用了cuda的版本,生成了对应的.so 文件

三、解决方案

1、重新安装cuda
2、删除原来的编译结果,重新进行编译

你可能感兴趣的:(深度学习,人工智能)