Segmentation fault on import with RTX 3090
I've just installed an RTX 3090 in my DL machine. I have the following dependencies installed
Driver Version: 455.32
Cuda Version: 11.0
Python Version: 3.8.6
Pytorch Version: 1.7.0+cu11  (pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html)
After initially having issues (installed cuda-11-1 rather than cuda-11-0) torch appears to work. I ran a  CIFAR10 example on GPU, I verified with nvidia-smi that it was indeed running on the GPU.
I installed the latest pytorch-lightning, i.e. pip install git+https://github.com/PytorchLightning/pytorch-lightning.git@master --upgrade but I cannot import, i.e.
>>> import pytorch_lightning as pl
Segmentation fault (core dumped)

Should this work? If so is there anything I've done wrong? If not, is there anything I can do to assist in tracking down the issue. Ironically I've just discovered that the CRF decoder is quicker on CPU (at least cf my older 1080Ti) so I don't need the GPU backend but atm I crashing on import is problematic ;)