Update README.md
Browse files
README.md
CHANGED
@@ -96,14 +96,14 @@ Note: You will need NVIDIA Ampere or Hopper GPUs to work with this model.
|
|
96 |
You will need to install NVIDIA Apex and NeMo.
|
97 |
|
98 |
```
|
99 |
-
git clone https://github.com/
|
100 |
cd apex
|
101 |
-
git checkout
|
102 |
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" --global-option="--fast_layer_norm" --global-option="--distributed_adam" --global-option="--deprecated_fused_adam" ./
|
103 |
```
|
104 |
|
105 |
```
|
106 |
-
pip install nemo_toolkit['nlp']==1.
|
107 |
```
|
108 |
|
109 |
Alternatively, you can use NeMo Megatron training docker container with all dependencies pre-installed.
|
|
|
96 |
You will need to install NVIDIA Apex and NeMo.
|
97 |
|
98 |
```
|
99 |
+
git clone https://github.com/NVIDIA/apex.git
|
100 |
cd apex
|
101 |
+
git checkout 03c9d80ed54c0eaa5b581bf42ceca3162f085327
|
102 |
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" --global-option="--fast_layer_norm" --global-option="--distributed_adam" --global-option="--deprecated_fused_adam" ./
|
103 |
```
|
104 |
|
105 |
```
|
106 |
+
pip install nemo_toolkit['nlp']==1.17.0
|
107 |
```
|
108 |
|
109 |
Alternatively, you can use NeMo Megatron training docker container with all dependencies pre-installed.
|