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.py from 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.
The current Dockerfile uses an image with outdated CUDA library
cudagl:11.3.0-develwhich 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.docker/dockerfile:bundlesdf/mycuda/setup.pyfrom c++14 to c++17:With these fixes, I was able to successfully build and run the provided examples:
run_demo.pyandbundlesdf/run_nerf.py.