PyTorch:从源代码生成TBB失败

Question

我试图使用TBB而不是OpenMP从源代码构建PyTorch，但每次都会出现以下错误：

    /usr/bin/ld: <PATH>/pytorch/build/lib/libtorch_cpu.so: undefined reference to `omp_get_max_threads'
    /usr/bin/ld: <PATH>/pytorch/build/lib/libtorch_cpu.so: undefined reference to `omp_get_num_threads'
    /usr/bin/ld: <PATH>/pytorch/build/lib/libtorch_cpu.so: undefined reference to `GOMP_barrier'
    /usr/bin/ld: <PATH>/pytorch/build/lib/libtorch_cpu.so: undefined reference to `GOMP_parallel'
    /usr/bin/ld: <PATH>/pytorch/build/lib/libtorch_cpu.so: undefined reference to `omp_in_parallel'
    /usr/bin/ld: <PATH>/pytorch/build/lib/libtorch_cpu.so: undefined reference to `omp_get_thread_num'
    collect2: error: ld returned 1 exit status
    make[2]: *** [caffe2/CMakeFiles/kernel_function_legacy_test.dir/build.make:109: bin/kernel_function_legacy_test] Error 1

您可以找到完整的日志文件HERE：

复制

复制行为的步骤：

1. git clone https://github.com/pytorch/pytorch
2. cd pytorch
3. ATEN_THREADING=TBB BUILD_BINARY=1 USE_EIGEN_THREADPOOL=1 USE_CUDA=0 PARALLEL_BACKEND=NATIVE_TBB USE_OPENMP=0 USE_TBB=1 MKL_THREADING=TBB BLAS=MKL USE_MKLDNN=1 MKLDNN_THREADING=TBB BUILD_BINARY=1 python setup.py build --cmake 2>&1 | tee ~/output.txt

• PyTorch版本（例如1.0）：1.10.0a0+git1798ff0
• 操作系统（如Linux）：Manjaro
• 如何安装PyTorch（conda，pip，源代码）：pip
• Python版本：3.9.5
• CUDA/cuDNN版本：否
• GPU型号和配置：否

Answer 1

似乎Caffe2 CMake文件存在问题。 您是否尝试过将CMake更新到最新版本。

在github上也可以找到类似的问题。通过升级cmake解决了问题

https://github.com/pytorch/pytorch/issues/8370