My env: Pytorch: 1.9.0+cu111
GPU: NVIDIA GeForce RTX 2080 Ti
Sys: Ubuntu 18.04.5 LTS
I ran python setup.py build
error: command 'gcc' failed with exit status 1
And no surprise, I encountered ModuleNotFoundError: No module named 'correlation_cuda' when I ran python demo.py
I tried:
sudo apt-get update
sudo apt-get install python-dev
sudo apt-get install python3-dev
sudo apt-get install libevent-dev
sudo apt-get install build-essential
sudo apt-get install libblas-dev libatlas-base-dev
sudo apt install gcc
But still not work.
Help me pls! Thx!