bigscience-workshop / bigscience-workshop/Megatron-DeepSpeed

Can I use python only apex for gpt_pretrain?

Open
#373 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
1.4k
Forks
226
PR merge metrics
No merged PRs in 30d

Description

I'm tring to run pretrain_gpt_single_node.sh in this [link](https://github.com/bigscience-workshop/Megatron-DeepSpeed/blob/main/examples/pretrain_gpt_single_node.sh), and find not amp_C installed, it needs gcc>=6.0 but I can't change gcc version for now. So will it possible to run this script without amp_C?

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.