The PyTorch binaries ship with the CUDA and cudnn runtimes, so you would just need to install the driver on your machine.
Which GPU(s) are you using?
Does nvidia-smi run successfully?
Did you update the driver without restarting?