PyTorch не использует CUDA ⇐ Python

[Anonymous]

У меня возникли проблемы при использовании PyTorch. CUDA настроен правильно, но по какой-то причине не используется PyTorch.
Да, проблема не нова, я видел много обсуждений на эту тему, но не понял ответ.
Я буду краток.
Сейчас я изучаю PyTorch, поэтому по сути я просто переписываю этот код, и у меня проблемы с CUDA: https://www.learnpytorch.io/03_pytorch_computer_vision/
Версия PyTorch+CUDA:

Код: Выделить всё

print(torch.__version__)
device = torch.device('cuda') if torch.cuda.is_available() else "cpu"
device
2.5.1+cu118
device(type='cuda')

Устройство доступно:

Код: Выделить всё

torch.cuda.is_available(), torch.cuda.device_count()
(True, 1)

Также я указал устройство (cuda) в данных и модели:

Код: Выделить всё

print(next(model_0.parameters()).device)
cuda:0

X, y = X.to(device), y.to(device)

Итак, в модели, обучающей чистое использование ЦП:


Свойства моего графического процессора, время обучения модели:

Код: Выделить всё

+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 561.00                 Driver Version: 561.00         CUDA Version: 12.6     |
|-----------------------------------------+------------------------+----------------------+
| 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 4060 ...  WDDM  |   00000000:01:00.0 Off |                  N/A |
| N/A   64C    P8              3W /   42W |     205MiB /   8188MiB |      0%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+

+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI        PID   Type   Process name                              GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|    0   N/A  N/A     19996      C   ...rograms\Python\Python310\python.exe      N/A      |
+-----------------------------------------------------------------------------------------+

Жду ваших советов и большое спасибо!

Подробнее здесь: https://stackoverflow.com/questions/792 ... using-cuda