Update Tutorial

Namhyeon Go 2024-06-15 12:48:45 +00:00
parent 9ba1e8712c
commit 548637ec36

@ -59,6 +59,26 @@
$ sudo rm -rf /usr/local/cuda*
```
* nvida 드라이버 설치
1. 설치 가능한 드라이버 찾기
```
$ sudo apt install ubuntu-drivers-common
$ sudo ubuntu-drivers devices # 버전 숫자가 높은 드라이버를 확인
```
2. 확인된 드라이버 설치
```
$ sudo apt install nvidia-driver-555-open
```
* cuda 설치
```
$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
$ sudo dpkg -i cuda-keyring_1.1-1_all.deb
$ sudo apt-get update
$ sudo apt-get -y install cuda
```
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