Pytorch For Cuda 12.6 !!hot!!

torch.backends.cudnn.benchmark = True

For users preferring the Anaconda or Miniconda ecosystem, you can specify the pytorch-cuda version: pytorch for cuda 12.6