[개발]run ai models using docker with gpu use

1. 허깅페이스에서 git-lfs 를 통해 원하는 ai model을 원하는 경로로 다운로드

2. docker를 설치

3. Dockerfile을 작성

# Use an official Pytorch runtime as the base image
FROM pytorch/pytorch:latest

# Set the working directory in the container
WORKDIR /app

# Install any needed packages specified in requirements.txt
COPY requirements.txt ./
RUN pip install --no-cache-dir -r requirements.txt

# Run the Python script when the container launches
CMD ["python", "script.py"]

4. requirements.txt를 작성

 

5. script.py를 작성

 

6. docker 명령을 통해 image를 생성하고 실행

#create docker image
$sudo docker build -t image_name .
#docker run
$sudo docker run --gpus all -v $(pwd):/app image_name

--gpus all 플래그를 통해 gpu 사용

 

누군가에게는 도움이 되길 바랍니다.