Support for RTX 4090 #27
Description
The current Dockerfile uses an image with outdated CUDA library cudagl:11.3.0-devel which does not support RTX 4090. To fix this, the docker image and torch version in the Dockerfile can be updated. Following changes worked for me.
- Update docker image and torch version in
docker/dockerfile:
FROM nvidia/cuda:12.1.0-devel-ubuntu20.04
...
RUN conda init bash &&\
echo "conda activate my" >> ~/.bashrc &&\
conda activate my &&\
pip install torchvision==0.16.0+cu121 torchaudio==2.1.0 torch==2.1.0+cu121 --index-url https://download.pytorch.org/whl/cu121 &&\
...
- Additionally, update the compiler flags in
bundlesdf/mycuda/setup.pyfrom c++14 to c++17:
nvcc_flags = ['-Xcompiler', '-O3', '-std=c++17', '-U__CUDA_NO_HALF_OPERATORS__', '-U__CUDA_NO_HALF_CONVERSIONS__', '-U__CUDA_NO_HALF2_OPERATORS__']
c_flags = ['-O3', '-std=c++17']
With these fixes, I was able to successfully build and run the provided examples: run_demo.py and bundlesdf/run_nerf.py.