Coding on ubuntu and I install pytorch via pip with specific cuda version like:

I need help please!
I need to use my GPU to perform CNN, but torch does not seem to detect my CUDA.

Im fairly new at anything related to python. Im trying to install CUDA for my GTX 1660. I installed Cuda Toolkit 12.5 first but then i downgraded it to 12.1. I still can’t get it to work.

I’m trying to run the VGG16 pre-trained model on GPU using Pytorch on Linux Mint. This code snippet

I’ve been working on optimizing the following PyTorch functions by rewriting them in Triton to speed up execution:

I tried to write a neural network with pytorch 1.8 cudatool 11.1 on RTX4090. But I met the following issues.

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.