[개발]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 containe..
기술과 IT
2023. 8. 22. 12:20