Update Tutorial
parent
9ba1e8712c
commit
548637ec36
20
Tutorial.md
20
Tutorial.md
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue
Block a user