So I tried implementing this repo: https://github.com/ntnu-ai-lab/SelfPAB on my computer,
I have downloaded the dataset but while i give train, the following issue comes:

I’m trying to get Pytorch working with my device which has CUDA 12 installed, and here are the packages installed:

I’m encountering an error when trying to use PyTorch with GPU acceleration. When I run torch.cuda.is_available(), it returns False.