Возобновить тренировку Caffe, используя другой GPU?

Question

Пожалуйста, прости, что может быть безумным вопросом.

Моя установка: две машины, одна с GTX1080, другая с RX Vega 64. Переподготовка / точная настройка модели bvlc_googlenet на GTX1080.

Если я собираю Caffe для Vega 64, могу ли я сделать снимок с машины GTX1080 и возобновить тренировки на Vega 64? Будет ли это работать в том смысле, что обучение будет продолжаться в обычном режиме?

Что если я перенес снимок GTX1080 на Volta V100 в AWS? Будет ли это работать?

Я знаю, что Кафе будет в некоторой степени абстрагировать аппаратное обеспечение, но я не знаю, насколько хорошо это может сделать это. Мне нужен GTX 1080 для чего-то еще ...

Заранее спасибо!

Answer 1

Насколько мне известно, это должно работать без проблем. Файлы веса и обучающие снимки - это просто набор чисел, которые вы сможете возобновить на другом оборудовании (например, CPU / GPU), на другом компьютере с другой операционной системой или между 32 и 64-разрядными процессами.