Cannot use other version of CUDA in installation

[xingzhehe]

I tried to use pip install "git+https://github.com/facebookresearch/pytorch3d.git" to install.

It tells me that

The detected CUDA version (10.2) mismatches the version that was used to compile
      PyTorch (11.2). Please make sure to use the same CUDA versions.

Indeed, I have CUDA 10.2 and CUDA 11.2 in /usr/local/, and 10.2 is the default.

After I used

export PATH=/usr/local/cuda-11.2/bin${PATH:+:${PATH}}
export PYTHONPATH=$PYTHONPATH:/usr/local/cuda-11.2/lib64/
export LD_LIBRARY_PATH=/usr/local/cuda-11.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

The nvcc -V shows the version is 11.2 now, but the original error is still there.

[bottler]

Can you set CUDA_HOME to "/usr/local/cuda-11.2" and try the build of pytorch3d again?