Проблема с 3D — свертками с PyTorch с использованием ROCM/MIOpen-Прямая свертка не может быть выполнена из-за неправильных параметров

Post author:admin
Запись опубликована:2 января, 2022
Post category:Вопросы по программированию

#pytorch #amd-rocm #пыторч #amd-rocm Вопрос: Я пытаюсь использовать nn.Conv3d PyTorch для сверточных автокодеров в системе с графическими процессорами AMD. У нас есть последние версии ROCM (4.5) и MIOpen (2.14). Тот…

Продолжить чтение