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

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

Сообщение Anonymous »

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

[*] gpu: < /strong> tesla k20m (вычислительные возможности 3.5)
[*] Общенациональная CUDA: [/b] cuda 11.2 установлен
[*] Компилятор: только GCC 9.2.0 доступен
[*] python: я пытался использовать Python 3.7 и 3.8 в различных средах Conda
текущие попытки: < /strong>

Я также пытался использовать различные комбинации предварительно построенных двоичных файлов и версий 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 или документация будут очень оценены.
Заранее спасибо за вашу помощь!

Подробнее здесь: https://stackoverflow.com/questions/794 ... m-35-on-an
Реклама
Ответить Пред. темаСлед. тема

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

  • Похожие темы
    Ответы
    Просмотры
    Последнее сообщение

Вернуться в «Python»