반응형
[도커파일 작성]
# 베이스 이미지 설정
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
# 타임존 설정
ENV TZ=Asia/Seoul
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
# workdir 변경 및 복사
WORKDIR /workspace/sam_annotator_v2
COPY . /workspace/sam_annotator_v2
# 필요 lib 설치
RUN apt-get update
RUN apt-get install -y nano
RUN apt-get install -y git
RUN apt-get install -y libgl1-mesa-glx
RUN apt-get install -y libglib2.0-0
# xcb 에러 해결
RUN apt-get install -y qt5-default libxcb-xinerama0-dev
# requriements 설치
RUN pip install -r requirements.txt
[도커파일 빌드]
docker build . -t <도커파일이름>
docker build . -t anno
[도커파일 런]
docker run -itd --gpus all --restart always --name anno_ -v /mnt/d/data:/usr/src/data anno /bin/bash
반응형
댓글