|
34 | 34 |
|
35 | 35 | # COMMAND ---------- |
36 | 36 |
|
37 | | -!wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcusparse-dev-11-3_11.5.0.58-1_amd64.deb -O /tmp/libcusparse-dev-11-3_11.5.0.58-1_amd64.deb && \ |
38 | | - wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcublas-dev-11-3_11.5.1.109-1_amd64.deb -O /tmp/libcublas-dev-11-3_11.5.1.109-1_amd64.deb && \ |
39 | | - wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcusolver-dev-11-3_11.1.2.109-1_amd64.deb -O /tmp/libcusolver-dev-11-3_11.1.2.109-1_amd64.deb && \ |
40 | | - wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcurand-dev-11-3_10.2.4.109-1_amd64.deb -O /tmp/libcurand-dev-11-3_10.2.4.109-1_amd64.deb && \ |
41 | | - dpkg -i /tmp/libcusparse-dev-11-3_11.5.0.58-1_amd64.deb && \ |
42 | | - dpkg -i /tmp/libcublas-dev-11-3_11.5.1.109-1_amd64.deb && \ |
43 | | - dpkg -i /tmp/libcusolver-dev-11-3_11.1.2.109-1_amd64.deb && \ |
44 | | - dpkg -i /tmp/libcurand-dev-11-3_10.2.4.109-1_amd64.deb |
| 37 | +# MAGIC !wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcusparse-dev-11-3_11.5.0.58-1_amd64.deb -O /tmp/libcusparse-dev-11-3_11.5.0.58-1_amd64.deb && \ |
| 38 | +# MAGIC wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcublas-dev-11-3_11.5.1.109-1_amd64.deb -O /tmp/libcublas-dev-11-3_11.5.1.109-1_amd64.deb && \ |
| 39 | +# MAGIC wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcusolver-dev-11-3_11.1.2.109-1_amd64.deb -O /tmp/libcusolver-dev-11-3_11.1.2.109-1_amd64.deb && \ |
| 40 | +# MAGIC wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcurand-dev-11-3_10.2.4.109-1_amd64.deb -O /tmp/libcurand-dev-11-3_10.2.4.109-1_amd64.deb && \ |
| 41 | +# MAGIC dpkg -i /tmp/libcusparse-dev-11-3_11.5.0.58-1_amd64.deb && \ |
| 42 | +# MAGIC dpkg -i /tmp/libcublas-dev-11-3_11.5.1.109-1_amd64.deb && \ |
| 43 | +# MAGIC dpkg -i /tmp/libcusolver-dev-11-3_11.1.2.109-1_amd64.deb && \ |
| 44 | +# MAGIC dpkg -i /tmp/libcurand-dev-11-3_10.2.4.109-1_amd64.deb |
45 | 45 |
|
46 | 46 | # COMMAND ---------- |
47 | 47 |
|
|
167 | 167 |
|
168 | 168 | # COMMAND ---------- |
169 | 169 |
|
170 | | -!deepspeed {num_gpus_flag} \ |
171 | | - --module training.trainer \ |
172 | | - --input-model {input_model} \ |
173 | | - --deepspeed {deepspeed_config} \ |
174 | | - --epochs 2 \ |
175 | | - --local-output-dir {local_output_dir} \ |
176 | | - --dbfs-output-dir {dbfs_output_dir} \ |
177 | | - --per-device-train-batch-size {batch_size} \ |
178 | | - --per-device-eval-batch-size {batch_size} \ |
179 | | - --logging-steps 10 \ |
180 | | - --save-steps 200 \ |
181 | | - --save-total-limit 20 \ |
182 | | - --eval-steps 50 \ |
183 | | - --warmup-steps 50 \ |
184 | | - --test-size 200 \ |
185 | | - --lr 5e-6 |
| 170 | +# MAGIC !deepspeed {num_gpus_flag} \ |
| 171 | +# MAGIC --module training.trainer \ |
| 172 | +# MAGIC --input-model {input_model} \ |
| 173 | +# MAGIC --deepspeed {deepspeed_config} \ |
| 174 | +# MAGIC --epochs 2 \ |
| 175 | +# MAGIC --local-output-dir {local_output_dir} \ |
| 176 | +# MAGIC --dbfs-output-dir {dbfs_output_dir} \ |
| 177 | +# MAGIC --per-device-train-batch-size {batch_size} \ |
| 178 | +# MAGIC --per-device-eval-batch-size {batch_size} \ |
| 179 | +# MAGIC --logging-steps 10 \ |
| 180 | +# MAGIC --save-steps 200 \ |
| 181 | +# MAGIC --save-total-limit 20 \ |
| 182 | +# MAGIC --eval-steps 50 \ |
| 183 | +# MAGIC --warmup-steps 50 \ |
| 184 | +# MAGIC --test-size 200 \ |
| 185 | +# MAGIC --lr 5e-6 |
186 | 186 |
|
187 | 187 | # COMMAND ---------- |
188 | 188 |
|
|
0 commit comments