
576 Глава 17
Рис. 17.6. Использование графических процессоров в Google Colab
ми пакетами — такими как NumPy, SciPy и последней стабильной версией PyTorch. На
момент подготовки книги последней стабильной версией в Google Colab была PyTorch 1.9.
Теперь мы можем протестировать рабочую среду и убедиться, что GPU доступен, ис-
пользуя следующий код:
>>> import torch
>>> print(torch.__version__)
1.9.0+cu111
>>> print("Доступность GPU:", torch.cuda.is_available())
Доступност ь GPU: True
>>> if torch.cuda.is_available():
... device = torch.device("cuda:0")
... else:
... device = "cpu"
>>> print(device)
cuda:0
Кроме того, если вы хотите сохранить ...