Добавить условие для выбора CPU/GPU
Сейчас модуль поиска узелков запускается только на GPU (cuda:0
) и на машинах, где нет cuda GPU
модуль выдает ошибку. Нужно добавить проверку доступности CUDA в yasx.py
(по умолчанию сделать cpu
).
Сейчас модуль поиска узелков запускается только на GPU (cuda:0
) и на машинах, где нет cuda GPU
модуль выдает ошибку. Нужно добавить проверку доступности CUDA в yasx.py
(по умолчанию сделать cpu
).