Now there is a problem with pytorch3d, because the direct PIP installation does not support GPU, so re install it

Re install command:

pip install “git+ https://github.com/facebookresearch/pytorch3d.git ”