鲲鹏社区首页
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助

编译过程中相关报错及解决方案

  • 报错一:不支持GPU架构,出现“nvcc fatal : Unsupported gpu architecture 'compute_35'”报错。

    可使用环境变量指定支持的GPU架构:

    1
    export TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6"
    
  • 报错二:由于内存不足,编译过程中系统资源被完全占用,导致服务器发生卡顿甚至停止响应。

    可限制并行任务数,如果内存不足可以适当减少数量:

    1
    export MAX_JOBS=32
    
  • 报错三:提示elsize找不到

    请使用NumPy 2.1以下版本

  • 报错四:引用torch可能会报错,报错信息如下图所示:

    此时需要退出当前的源码目录,可以切换到一个无关的目录,例如:/tmp,然后在新的目录下再次运行python3import torch指令。