RTX 3070 compatibility with Pytorch

NVIDIA GeForce RTX 3070 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.

So I'm currently trying to train a neural network but I'm getting this issue. It seems that the GPU model I have is not compatible with the version of PyTorch that I have.
The output of my nvcc -V is:
Cuda compilation tools, release 11.4, V11.4.48
Build cuda_11.4.r11.4/compiler.30033411_0

and my PyTorch version is 1.9.0.
I've tried changing the CUDA version from what was initially 10 to 11.4 and there was no change. Any help would be hugely appreciated.

It might be because you have installed a torch package with cuda==10.* (e.g. torch==1.9.0+cu102) . I'd suggest trying:
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

