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

>conda list torch
# packages in environment:
#
# Name                    Version                   Build  Channel
pytorch                   2.3.0           py3.11_cuda12.1_cudnn8_0    pytorch
pytorch-cuda              12.1                 hde6ce7c_5    pytorch
pytorch-mutex             1.0                        cuda    pytorch
torchaudio                2.3.0                    pypi_0    pypi
torchvision               0.18.0                   pypi_0    pypi

The pytorch version is clearly for CUDA devices. However, on print(torch.__version__); torch.cuda.is_available()

I get

2.3.0
False