Блог

Необходимо ли преобразовывать тензоры и модели в CUDA с tensor.to помощью in colab, когда я выбрал тип среды выполнения в качестве GPU?

Я хочу использовать CUDA для обучения своей модели

• tensor.to(устройство) передача данных на данное устройство.
• Да, вам нужно перенести модель, ввод, метки и т.д. На любое устройство, которое вы собираетесь использовать