I need to install a CUDA version of a package (python)

Job ID: 31402677

Budget: $10 – $30 USD

I have a code from Github which I want you to help me run successfully. I have installed all requirements of packages needed but when i run the code, I get an error saying a CUDA version of a torch package is not installed.

When i run torch.cuda.is_available() i get TRUE and torch.cuda.current_device() returns 0, and torch.cuda.device_count() returns 1.

This proves it is not a software problem with CUDA but it is an environmental issue, so I think I need to download another way?
Related categories: Python Software Architecture CUDA