Install pytorch cuda 11.4 with conda

1

Hello,
I tried to download PyTorch that matches my device:
Driver Version: 470.199.02
CUDA Version: 11.4
Ubuntu: 20.04.4
However, I cannot find the Pytorch version for CUDA 11.4 on the PyTorch official website. Most of the information related to CUDA 11.4 on the internet is kind of outdated.

Also, I tried torch==2.5.1, I tried to run torch.cuda.is_available() and the output is False. And the output of torch.cuda.device_count() is 1. Anyone has the idea of how to fix this problem? Thank you!!

2

I would not recommend trying to install such old PyTorch releases. Instead try to install the latest one with CUDA 11.8 or update your NVIDIA driver and install the latest PyTorch binary with any CUDA 12 runtime dependency.