Hi, I cant get pytorch to download the GPU version. I have tried multiple times and am coming up with the same issue.
±----------------------------------------------------------------------------------------+
| NVIDIA-SMI 552.44 Driver Version: 552.44 CUDA Version: 12.4 |
|-----------------------------------------±-----------------------±---------------------+
| GPU Name TCC/WDDM | 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 2060 WDDM | 00000000:01:00.0 On | N/A |
| N/A 41C P8 7W / 80W | 62MiB / 6144MiB | 0% Default |
| | | N/A |
±----------------------------------------±-----------------------±---------------------+
±----------------------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=========================================================================================|
±----------------------------------------------------------------------------------------+
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Thu_Mar_28_02:30:10_Pacific_Daylight_Time_2024
Cuda compilation tools, release 12.4, V12.4.131
Build cuda_12.4.r12.4/compiler.34097967_0
and in python I run a script and get
AssertionError: Torch not compiled with CUDA enabled
CUDA Available: False
PyTorch version: 2.3.0+cpu
CUDA Version: None
Number of GPUs: 0
I tried installing: pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu124
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121
All with the same outcome