Sure!
Check the install instructions here 1 and select “CUDA 11.3”:

image843×325 22.7 KB

Make sure to uninstall the current PyTorch build in the used environment (run pip uninstall torch -y a few times and/or conda uninstall pytorch -y) or create a new virtual environment before installing the new binaries.