lkeab / BCNet

Deep Occlusion-Aware Instance Segmentation with Overlapping BiLayers [CVPR 2021]
https://arxiv.org/abs/2103.12340
MIT License
529 stars 74 forks source link

RuntimeError: No such operator torchvision::nms #133

Open ZNan-Chen opened 1 year ago

ZNan-Chen commented 1 year ago

Hello, I followed the instructions to install the environment and ran into this problem: RuntimeError: No such operator torchvision::nms

I am using 2080Ti, here is my environment information:

_libgcc_mutex 0.1 main https://mirrors.ustc.edu.cn/anaconda/pkgs/main _openmp_mutex 5.1 1_gnu https://mirrors.ustc.edu.cn/anaconda/pkgs/main _pytorch_select 0.2 gpu_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main blas 1.0 mkl https://mirrors.ustc.edu.cn/anaconda/pkgs/main ca-certificates 2023.5.7 hbcca054_0 conda-forge certifi 2023.5.7 pyhd8ed1ab_0 conda-forge cffi 1.15.0 py37h036bc23_0 conda-forge cloudpickle 2.0.0 pypi_0 pypi cudatoolkit 10.0.130 h8c5a6a4_10 conda-forge cudnn 7.6.5.32 ha8d7eb6_1 conda-forge cycler 0.11.0 pypi_0 pypi cython 0.29.24 pypi_0 pypi detectron2 0.1 pypi_0 pypi future 0.18.3 pypi_0 pypi imageio 2.27.0 pypi_0 pypi intel-openmp 2023.1.0 hdb19cb5_46305 https://mirrors.ustc.edu.cn/anaconda/pkgs/main iopath 0.1.10 pypi_0 pypi kiwisolver 1.4.4 pypi_0 pypi ld_impl_linux-64 2.38 h1181459_1 https://mirrors.ustc.edu.cn/anaconda/pkgs/main libffi 3.4.4 h6a678d5_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main libgcc-ng 11.2.0 h1234567_1 https://mirrors.ustc.edu.cn/anaconda/pkgs/main libgomp 11.2.0 h1234567_1 https://mirrors.ustc.edu.cn/anaconda/pkgs/main libstdcxx-ng 11.2.0 h1234567_1 https://mirrors.ustc.edu.cn/anaconda/pkgs/main matplotlib 3.4.3 pypi_0 pypi mkl 2020.2 256 https://mirrors.ustc.edu.cn/anaconda/pkgs/main mkl-service 2.3.0 py37h8f50634_2 conda-forge mkl_fft 1.2.0 py37h161383b_1 conda-forge mkl_random 1.2.0 py37h9fdb41a_1 conda-forge ncurses 6.4 h6a678d5_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main networkx 2.6.3 pypi_0 pypi ninja 1.10.2.2 pypi_0 pypi numpy 1.21.6 pypi_0 pypi opencv-python 4.4.0.40 pypi_0 pypi openssl 1.1.1t h7f8727e_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main pillow 6.2.2 pypi_0 pypi pip 22.3.1 py37h06a4308_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main portalocker 2.5.1 pypi_0 pypi pycocotools 2.0 pypi_0 pypi pycparser 2.21 pyhd8ed1ab_0 conda-forge pydot 1.4.2 pypi_0 pypi pyparsing 3.0.9 pypi_0 pypi python 3.7.16 h7a1cb2a_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main python-dateutil 2.8.2 pypi_0 pypi python_abi 3.7 2_cp37m conda-forge pytorch 1.4.0 cuda100py37h75a84eb_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main pywavelets 1.3.0 pypi_0 pypi pyyaml 6.0 pypi_0 pypi readline 8.2 h5eee18b_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main scikit-image 0.18.3 pypi_0 pypi scipy 1.7.3 pypi_0 pypi setuptools 65.6.3 py37h06a4308_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main six 1.16.0 pyh6c4a22f_0 conda-forge sqlite 3.41.2 h5eee18b_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main tabulate 0.8.10 pypi_0 pypi termcolor 1.1.0 pypi_0 pypi tifffile 2021.11.2 pypi_0 pypi tk 8.6.12 h1ccaba5_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main torchvision 0.5.0 pypi_0 pypi tqdm 4.62.3 pypi_0 pypi typing-extensions 4.5.0 pypi_0 pypi wheel 0.38.4 py37h06a4308_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main xz 5.4.2 h5eee18b_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main yacs 0.1.8 pypi_0 pypi zlib 1.2.13 h5eee18b_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main