git clone https://github.com/pytorch/vision cd vision 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