Failed to build wheel in Nvidia CUDA 10.1 Ubuntu 16.04 Docker image

[MiroPsota]

Instructions To Reproduce the Issue:

Build Dockerfile and run commands in build_wheel_pytorch3d.sh.

Relevant error:

Parts with errors in error.log and full log in full.log.

Notes:

I can build pytorch3d wheel with cuda 10.1 Ubuntu 18.04 and cuda 11.0 Ubuntu 20.04 as base images so it's likely some incompatibility with GCC 5.4?
Also buildable with cuda 10.1 Ubuntu 16.04 at commit dc08c30.

[bottler]

This is the problem with cloneable not working in nvcc. It can be fixed by making a change like 85c396f to the code recently added in bce396d.

With pytorch master I think this won't be a problem because of a fix like pytorch/pytorch#37221. It also might be working with latest pytorch (1.5.1) rather than the 1.5.0 which you might be picking up - I'm not sure about this, but I suspect that's why our CI is happy at the moment.

[MiroPsota]

I confirm it's building with pytorch 1.5.1. I suggest to add note to INSTALL.md.