Pytorch vrsion for cuda 12.7

1

Newb question. Im new to machine learning and Im trying to install pytorch. What is the compatible version for cuda 12,7?
±----------------------------------------------------------------------------------------+
| NVIDIA-SMI 566.36 Driver Version: 566.36 CUDA Version: 12.7 |
|-----------------------------------------±-----------------------±---------------------+
| GPU Name Driver-Model | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|=========================================+========================+======================|
| 0 NVIDIA GeForce RTX 3050 … WDDM | 00000000:01:00.0 Off | N/A |
| N/A 53C P3 5W / 38W | 221MiB / 4096MiB | 5% Default |
| | | N/A |
±----------------------------------------±-----------------------±---------------------+

±----------------------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=========================================================================================|
| 0 N/A N/A 20256 C …a09mmv6hy\Build\Plugins\Mpv\mpv.exe N/A |
| 0 N/A N/A 20440 C …al\Discord\app-1.0.9174\Discord.exe N/A |
±----------------------------------------------------------------------------------------+

2

You are referring to the driver (566.36), which is new enough to support all of our PyTorch binaries (up the he newest CUDA toolkit 12.6 available as nightly binaries).

3

Dear developers, it is real pain in ass to find actual installation info. Anaconda, CUDA Toolkit, cuDNN, environment variables, to compile something, %n%???

When my way was pretty simple:
Win11 x64
nvidia-smi gives to me cuda 12.7
I installed CUDA Toolkit 12.6 (do I really need it?)
Used this command “pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126”
And command “torch.cuda.is_available()” gives to me True, finally

4 Likes
4

I don’t see a question in your post and it seems your setup works?

5

So thankful when I search something and someone posts the answer 3 days earlier! TY!!!

6

PyTorch comes with pre-compiled cuda code so you do not need to download Nvidia Cuda toolkit or Cudnn

7

thank you so much it works for me 2 hour searching finally i got it