Pytorch For Cuda 12.6 ((free)) -

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

python setup.py develop

| Component | Version | |-----------|---------| | OS | Ubuntu 22.04 / 24.04, or Windows 11 / Server 2022 | | NVIDIA Driver | 550.54.15+ (supports CUDA 12.6) | | CUDA Toolkit | 12.6.0+ | | cuDNN | 9.3.0+ (for CUDA 12.x) | | Python | 3.9 – 3.12 | | PyTorch | 2.5.0+ (or source build) | pytorch for cuda 12.6

nvcc --version nvidia-smi # Expected: CUDA Version 12.6 pytorch for cuda 12.6