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

一、 问题描述

ImportError: /home/jankin/.cache/torch_extensions/py38_cu117/fused/fused.so: cannot open shared object file: No such file or directory

二、问题分析

在进行编译,且编译成功,
运行main.py后,还出现此问题

三、解决方案

使用conda重新创建一个环境进行编译,编译成功。

  • 注意:在运行main函数时生成fused.so 文件

你可能感兴趣的:(python)