Ubuntu16.04從源碼安裝detectron2

準備工作

首先確保pytorch的cuda版本和系統的一致。

首先查看pytorch對應的cuda版本，通過conda list命令可以查看，我這邊裝的是1.5，對應的是cuda-10.1

pytorch 1.5.0 py3.7_cuda10.1.243_cudnn7.6.3_0    pytorch

再查看系統的cuda版本，即/usr/local/路徑下cuda文件夾對應的cuda版本，我這邊是下面這樣

cuda-10.0
cuda-9.0
cuda-8.0
cuda #軟連接，指向cuda-10.0

系統沒有cuda-10.1，可以從官網下載或者從其他電腦拷貝一份放在這里，然后將軟連接指向cuda-10.1，命令如下

cd /usr/local/
rm -rf cuda
ln -s /usr/local/cuda-10.1 cuda

安裝

然后按照官網給的命令，從源碼安裝即可。
命令如下：

git clone https://github.com/facebookresearch/detectron2.git
cd detectron2
python setup.py build develop

安裝完之后，通過命令

ldd build/lib.linux-x86_64-3.7/detectron2/_C.cpython-37m-x86_64-linu
x-gnu.so

結果顯示如下：

        linux-vdso.so.1 =>  (0x00007ffe967f8000)
        libc10.so => not found
        libtorch.so => not found
        libtorch_cpu.so => not found
        libtorch_python.so => not found
        libcudart.so.10.1 => /opt/anaconda3/envs/pytorch-1.5/lib/libcudart.so.10.1 (0x00007fa99a4ce000)
        libc10_cuda.so => not found
        libtorch_cuda.so => not found
        libstdc++.so.6 => /opt/anaconda3/envs/pytorch-1.5/lib/libstdc++.so.6 (0x00007fa99a35a000)
        libm.so.6 => /lib/x86_64-linux-gnu/libm.so.6 (0x00007fa99a051000)
        libgcc_s.so.1 => /opt/anaconda3/envs/pytorch-1.5/lib/libgcc_s.so.1 (0x00007fa99a827000)
        libpthread.so.0 => /lib/x86_64-linux-gnu/libpthread.so.0 (0x00007fa999e34000)
        libc.so.6 => /lib/x86_64-linux-gnu/libc.so.6 (0x00007fa999a6a000)
        /lib64/ld-linux-x86-64.so.2 (0x00007fa99a74d000)
        libdl.so.2 => /lib/x86_64-linux-gnu/libdl.so.2 (0x00007fa999866000)
        librt.so.1 => /lib/x86_64-linux-gnu/librt.so.1 (0x00007fa99965e000)

如果出現錯誤

input, roi, spatial_scale, output_size[0], output_size[1], sampling_ratio, aligned
RuntimeError: CUDA error: invalid device function
Segmentation fault (core dumped)

可能是系統的cuda版本和pytorch的不一致，我系統cuda軟連接指向cuda-10.0時會報這個錯誤。

參考

• detectron2-INSTALL