Как решить проблему с Cuda на Google Colab?

Question

[coderKerk]

Сейчас начинаю изучение ИИ. Пишу на гугл колабе, так как у самого дискретной видюхи нет, что плохо бы сказалось. Использую Python, PyTorch. Когда создаю модель model = model.cuda() вылезает ошибка: RuntimeError: No CUDA GPUs are available. Как можно пофиксить?

Answer 1

[sswwssww]

Среда выполнения > сменить среду выполнения > аппаратный ускоритель > gpu

Comments

[coderKerk]

есть ли вариант без смены среды выполнения?

[sswwssww]

coderKerk,
попробуй так:

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
model = model.to(device)

[coderKerk]

sswwssww, спасибо огромное! Помогло!