Torch version + cuda stable compatibility details:

Mohan_Krishna · February 25, 2025, 2:21pm

I have installed NVIDIA-SMI 550.120 Driver Version: 550.120 CUDA Version: 12.4

$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Fri_Jan__6_16:45:21_PST_2023
Cuda compilation tools, release 12.0, V12.0.140
Build cuda_12.0.r12.0/compiler.32267302_0
I want a stable versions of torch and cuda compatibility for NVIDIA RTX A5000

which version should i proceed on cuda toolkit, torch, driver

ptrblck · February 25, 2025, 2:39pm

Your locally installed CUDA toolkit won’t be used as PyTorch binaries ship with their own CUDA runtime dependencies. You would need to install an NVIDIA driver first and can install any binaries afterwards. To install the latest stable version with CUDA support run pip install torch.

Mohan_Krishna · February 25, 2025, 2:57pm