@ptrblck I have the following error with the CUDA11.4
NVIDIA GeForce RTX 3090 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_61 sm_70 sm_75 compute_37.

I installed PyTorch using this command:
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

I would be grateful if you please point me in the right direction. Should I install Nightly version or docker?