Install pytorch with CUDA 11

The binaries are not built yet and you would have to install PyTorch from source at the moment.

CC @Santhosh_Kumar1, @bear_sun

2 Likes