From 36668dde72fd63dd104567b4c9bafb305fe4148f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=ED=95=9C=EC=B9=98=EC=98=81?= Date: Mon, 9 Dec 2024 14:48:09 +0900 Subject: [PATCH] =?UTF-8?q?=EC=98=A4=ED=83=80=20=EC=88=98=EC=A0=95.=20READ?= =?UTF-8?q?ME=20=EC=97=85=EB=8D=B0=EC=9D=B4=ED=8A=B8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 11 +++++++++++ docker-compose.yml | 2 +- 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 29e3f7a..3593fed 100644 --- a/README.md +++ b/README.md @@ -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 인식을 확인한다. \ No newline at end of file diff --git a/docker-compose.yml b/docker-compose.yml index bb87eea..2cba89b 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -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