运行python setup.py build install报错

部署运行你感兴趣的模型镜像

报错信息
报错如上:
/usr/include/wchar.h(582): error: attribute “malloc” does not take arguments

15 errors detected in the compilation of “/home/pytorch/InImage/ops_dc3/src/cuda/dcnv3_cuda.cu”.
error: command ‘/usr/local/cuda-11.5/bin/nvcc’ failed with exit code 1

还有报错:
RuntimeError:
The detected CUDA version (11.7) mismatches the version that was used to compile
PyTorch (10.2). Please make sure to use the same CUDA versions.

原因:g++是11.3版本,版本不匹配
https://discuss.pytorch.org/t/detected-cuda-version-11-7-mismatches-the-version-that-was-used-to-compile-pytorch-10-2/153348
解决方法:按https://blog.youkuaiyun.com/tangjiahao10/article/details/125501412的方法把g++降到10版本

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

### Python项目构建和安装过程中的常见错误及其解决方案 #### 构建与安装流程概述 `python setup.py install` 这条命令实际上已经包含了 `build` 和 `install` 的操作,可以一次性完成整个安装过程[^1]。然而,在某些情况下,可能希望先单独执行构建步骤再进行安装,以便于分发或调试。 #### 常见问题及解决方法 ##### 缺少依赖项 如果在运行 `python setup.py build` 或者 `python setup.py install` 时报错,可能是由于缺少必要的编译工具或其他依赖库造成的。对于C/C++扩展模块来说尤其如此,因为这些通常需要特定版本的编译器支持以及相应的头文件和静态链接库。 - **Linux环境** 对于基于Debian系统的发行版(如Ubuntu),可以通过apt-get来安装所需的开发工具链: ```bash sudo apt-get update && sudo apt-get upgrade -y sudo apt-get install build-essential python-dev default-libmysqlclient-dev libssl-dev ``` - **Windows环境** Windows环境下建议使用Visual Studio Community Edition提供的MSVC编译器,并确保已正确配置PATH变量指向vcvarsall.bat脚本的位置。 ##### 不兼容的操作系统平台 有时源码级发布的Python包仅适用于特定操作系统架构下的CPU指令集。例如尝试在一个不被官方支持平台上编译PyLorcon时可能会遭遇失败[^3]。此时应该寻找预编译二进制发布版本或是考虑更换目标机器类型。 ##### CUDA相关错误处理 当涉及到GPU加速计算框架(像PyTorch)的时候,还需要额外注意CUDA驱动程序及相关组件是否匹配当前使用的硬件设备型号。任何细微差异都可能导致编译阶段出现致命性的语法解析异常[^5]。 针对上述提到的情况,可采取如下措施加以应对: - 更新至最新稳定版CUDA Toolkit; - 修改Makefile或者setup.cfg里的宏定义参数以适应不同版本间的API变更; - 参考官方文档调整环境变量设置,特别是关于路径映射的部分; ```python import os os.environ['CUDA_HOME'] = '/usr/local/cuda' ``` #### 示例代码片段展示如何自定义Setup Configuration 为了更好地控制项目的打包行为,可以在`setup()`函数内部指定更多选项来自定义化构建逻辑: ```python from setuptools import setup, Extension module_name = 'example_module' extensions = [ Extension( name=module_name, sources=['src/example.cpp'], include_dirs=["./include"], libraries=[], extra_compile_args=[ '-std=c++17', '-O3', '-Wall' ] ) ] setup( name='Example Package Name', version='0.1dev', ext_modules=extensions, cmdclass={ "build_ext": BuildExtCommand } ) ```
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值