'Ensure pytorch with CUDA support is installed' - Error with running

[Mike-The-Knight]

I have run into a problem where I install CUDA 12.1 which fits my Graphics card, I then install pytorch for 12.1 from pytorch documentation to a conda environment. I then install ninja and install a requirements file for a separate project which includes pip installing tiny-cuda-nn and everything works fine.

However when I run the main file for the other project that uses tiny-cuda-nn it doesn't recognise my pytorch with CUDA support is installed or something along those lines.

Please help and I am happy to offer any extra information needed.

[gaetan-landreau]

+1
Got a similar issue while both my torch and CUDA drivers are properly installed. GPUs are detected, running pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch works like a charm, but the import fails.