오타 수정. README 업데이트

This commit is contained in:
2024-12-09 14:48:09 +09:00
parent 403ec561a1
commit 36668dde72
2 changed files with 12 additions and 1 deletions

View File

@@ -39,6 +39,8 @@ torch-base-boilerplate
### 빌드
' docker compose build '
> dockerfile상 변화가 없으나 디펜던시 변화 등으로 재빌드가 필요할 때는 --no-cache 옵션을 주어 캐시를 이용하지 않고 재빌드를 할 수 있도록 한다.
### 실행
' docker compose up '
@@ -62,3 +64,12 @@ sudo usermod -aG docker $USER
```
* WSL상에 설치는 VSCode로 WSL에 리모트 연결한 상태에서 Extension 설치로도 가능하다
### GPU 이용하기
```
$ curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
&& curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
$ sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
```
> $nvidia-smi 로 GPU 인식을 확인한다.

View File

@@ -8,7 +8,7 @@ services:
volumes:
- ./workspace:/opt/workspace
- ./cache:/root/.cache/
working_dir: /opt/whisper
working_dir: /opt/workspace
entrypoint: uvicorn api:app --host 0.0.0.0 --port 7890 --reload
restart: always