我正在使用 CMake 构建扩展
setup(
...
cmdclass={'build_ext':Cmakebuildclass}
)
其中
Cmakebuildclass
是自定义类。
在此类的 run() 函数内部,我正在构建扩展
它创建一个临时构建目录,其中存储所有工件。
然而,最后我尝试访问它但失败了
error: [Errno 2] No such file or directory: 'build/temp.linux-x86_64-3.6
临时构建目录[例如目标构建后,
build/temp.linux-x86_64-3.6
] 将被删除。
它通常被复制到构建中的 lib 文件夹
build/lib.linux-x86_64-3.6/
所以使用 build_lib 变量而不是 build_temp
我通过强制安装torch解决了这个问题,即在“pip3 install torch torchvision torchaudio”之后添加“--force” 记得检查cuda版本