1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
function init_torch() {
    # export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
    # echo ${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
    
    export CMAKE_BUILD_TYPE=debug
    export _GLIBCXX_USE_CXX11_ABI=1

    # export USE_LEVELDB=0
    # export USE_LMDB=0
    # export USE_OPENCV=0
    # export USE_TENSORRT=0
    # export USE_FFMPEG=0
    # export USE_REDIS=0

    # export USE_XNNPACK=0
    # export USE_XNNPACK=OFF

    # export USE_QNNPACK=0
    # export USE_NNPACK=0
    # export USE_NNPACK=OFF
    # export USE_MKLDNN=0
    # export USE_NNPACK=0
    # export USE_NINJA=0
    # export USE_PYTORCH_QNNPACK=0
    # export USE_PYTORCH_QNNPACK=OFF
    # export USE_ROCM=0
    # export USE_FBGEMM=0
    # export USE_OPENMP=0
    # export USE_NINJA=OFF
    # export USE_NINJA=0

    # export USE_OPENBLAS=0
    # export USE_OPENBLAS=OFF

    export BUILD_TEST=0
    export BUILD_TEST=OFF

    # export USE_CUDA=0
    # export USE_CUDA=OFF
    export USE_DISTRIBUTED=ON
    export CCACHE_DISABLE=1

    # cd build && rm -rf ./* && cd ../
}

cd pytorch
init_torch
BUILD_BINARY=0 USE_PRECOMPILED_HEADERS=1 BUILD_TEST=0  DEBUG=1 python setup.py build_ext -i

1
2
3
4
5
6
7
8
Python 3.8.10 (default, Jun 22 2022, 20:18:18)
[GCC 9.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> quit()
pure virtual method called
terminate called without an active exception
Aborted