Unable to run on a modern GPU (RTX A6000)

Running the software on an RTX A6000 GPU throws the following errors:

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.
If you want to use the NVIDIA RTX A6000 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
...
RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
Codec/Encoder: name=JAI, version=DIS

Is it a matter of updating PyTorch or are there some custom kernels that need a specific torch/CUDA version to run? Thanks in advance, I may provide some more information on my system if it's helpful.