Pip installer doesn't seem to work for A6000 GPU on Linux

I get this error and then it bugs:

`UserWarning: `
`NVIDIA RTX A6000 with CUDA capability sm_86 is not compatible with the current PyTorch installation.`
`The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.`

You are most likely installing the PyTorch binaries with the CUDA 10.2 runtime, while your Ampere GPU needs CUDA 11. Select the CUDA 11.3 or 11.6 runtime in the install box, use the created install command, and it should work.