From 57d583c97f64ed93220a23971c10a08350b51385 Mon Sep 17 00:00:00 2001 From: Namhyeon Go Date: Sat, 15 Jun 2024 12:56:09 +0000 Subject: [PATCH] Update Tutorial --- Tutorial.md | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/Tutorial.md b/Tutorial.md index 9c4d36d..dd02de1 100644 --- a/Tutorial.md +++ b/Tutorial.md @@ -79,6 +79,17 @@ $ sudo apt-get -y install cuda ``` + * nvidia 모듈 다시 로드 + ``` + $ sudo rmmod nvidia_drm + $ sudo rmmod nvidia_modeset + $ sudo rmmod nvidia_uvm + $ sudo rmmod nvidia + $ nvidia-smi + ``` + + 설치된 그래픽 카드가 정상적으로 표시되는지 확인. + 5. 선행학습(pretrain) 진행 ``` $ accelerate launch --config_file configs/accelerate_configs/ds_stage1.yaml train_lm.py --train_config configs/pretrain_config.yaml --model_config configs/model_configs/7B.json