You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
В файле requirements.txt лежат библиотеки, которые система устанавливает для корректной работы. В этом файле нас интересует строка torch. Если зайти а официальный сайт https://pytorch.org/get-started/locally/ увидим, что версия для линукса в базовой версии поддерживает ядра CUDA, а версия для Windows не поддерживает. Для установки пакета с поддержкой в Windows ядер CUDA замените строчку эту строчку на torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121. После этого переустановите контейнер и модель сможет запуститься на видеокарте.
Возможна проблема с совместимостью последней версии torch cдругими библиотеками, по той причине, что в проекте использовалась предыдущая версия. В таком случае укажите, что надо установить версию (1.7.1).
Не работает на gpu 3060ti на win10.
The text was updated successfully, but these errors were encountered: