Ищу предварительно построенную конфигурацию Pytorch, поддерживающую Tesla K20M (SM_35) в системе института ⇐ Python

[Anonymous]

Я пытаюсь обучить языковую модель в стиле GPT-2 в системе в моем институте, в которой используется графический процессор Tesla K20M (вычислительные возможности 3.5). < /p>
Я изначально пытался сделать Создайте Pytorch из источника, чтобы включить правильную поддержку ядра CUDA для моего GPU, установив переменную среды TORCH_CUDA_ARCH_LIST = "3,5" . Тем не менее, я сталкиваюсь с несколькими проблемами во время строительства из источника. Поэтому я бы предпочел использовать предварительно построенную версию Pytorch (наряду с соответствующей версией Cuda Toolkit и Python), которая явно поддерживает вычислительные возможности 3.5. /strong> < /p>

[*] gpu: < /strong> tesla k20m (вычислительные возможности 3.5)
[*] Общенациональная CUDA: [/b] CUDA 11.2 установлен
python: i Пытались использовать Python 3.7 и 3.8 в различных средах Conda < /li>
Попытки текущих Попытка использовать различные комбинации предварительно построенных двоичных файлов и версий Python, но они не включают необходимые ядра CUDA для SM_35, что приводит к ошибкам выполнения, таким как:
RuntimeError: CUDA error: no kernel image is available for execution on the device
< /code>
< /li>
< /ul>
< /li>
< /ul>
Мой вопрос: < /strong>

Есть ли рекомендуемая предварительно построенная конфигурация (т.е. конкретные версии Pytorch, Cuda Toolkit и Python), известный для поддержки Tesla K20M (Compute Paliebability 3.5) вне коробки? Предпочтительно, мне нужно решение, которое работает с общеобразовательной установкой CUDA (даже если это означает использование более старой версии CUDA, такой как 10.2), и которая позволяет избежать необходимости строительства из источника с обновленным GCC. Любые указатели на известные конфигурации, каналы Conda или документация будут очень оценены. < /P>
Заранее спасибо за вашу помощь! Мой институт для установки GCC 9.5.0, но сам оставшийся процесс строительства очень подвержен ошибке. Я пытаюсь перемещаться по нему.

Подробнее здесь: https://stackoverflow.com/questions/794 ... m-35-on-an