Как подключить видеокарту?

Question

[nano_e_t_4]

Хочу с коллаба перейти на локальную машину, у меня ноутбук lenovo legion с видеокартой Nvidia GeForce RTX 3070.
Как подключить ее в pycharm для обучения моделей?
Работают на windows 11 pro.

Answer 1

[dim5x]

1. Поставить CUDA.
2. Проверить:

import torch
torch.cuda.is_available()

3. Перенести модель на GPU:

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(device)
model = MyModel(args)
model.to(device)

От IDE не зависит. Torch Start Locally.