Open anas-zafar opened 2 years ago
RTX 3080 is Ampere architecture (https://arnon.dk/matching-sm-architectures-arch-and-gencode-for-various-nvidia-cards/) and requires CUDA 11.x I'm also looking to find a solution that will work with Ampere cards.
Here is the solution: Use Ubuntu 20.04 with CUDA 11.3.1, e.g. with Nvidia's docker container: docker://nvidia/cuda:11.3.1-devel-ubuntu20.04
Install system tools and dependencies:
apt install -y --no-install-recommends \
wget \
nano \
python3-setuptools \
python3-dev \
python3-pip \
cython \
git \
libx264-dev \
libglibd-2.0-dev \
libsm6 \
libxext6 \
libxrender-dev
Install Python dependencies without the version numbers given in the original requirements.txt!
pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113
pip3 install absl-py attrs cachetools certifi chardet click cycler Cython flake8 flake8-import-order future google-auth google-auth-oauthlib grpcio idna importlib-metadata iniconfig kiwisolver Markdown matplotlib mccabe motmetrics numpy oauthlib opencv-python packaging pandas Pillow pluggy protobuf py py-cpuinfo pyasn1 pyasn1-modules pycocotools pycodestyle pyflakes pyparsing pytest pytest-benchmark python-dateutil pytz PyYAML requests requests-oauthlib rsa scipy Shapely sheen six tabulate tensorboard tensorboard-plugin-wit tensorboardX termcolor toml tqdm urllib3 Werkzeug xmltodict yacs zipp gdown
Ignore the error messages about version incompatibilities. These are about packages that are probably not used anyway.
Patch two files to use new function call:
cd thirdparty/deform_conv
sed -i -e 's@AT_CHECK@TORCH_CHECK@' src/deform_conv_cuda.cpp
sed -i -e 's@AT_CHECK@TORCH_CHECK@' src/deform_pool_cuda.cpp
And then install as before.
Thank you for the amazing work! I am having an issue while installing DCN
I am using python 3.6.12, pytorch-1.4.0, and CUDA 10.0
My GPU Specifications: GeForce RTX 3080
Windows System
Any solutions for this?