| FROM anibali/pytorch:2.0.0-cuda11.8-ubuntu22.04 | |
| # Set up time zone. | |
| ENV TZ=UTC | |
| RUN sudo ln -snf /usr/share/zoneinfo/$TZ /etc/localtime | |
| USER root | |
| RUN apt update && apt install -y gcc-10 g++-10 && ln /usr/bin/gcc-10 /usr/bin/gcc && ln /usr/bin/g++-10 /usr/bin/g++ && apt install -y zlib1g-dev && rm -r /var/lib/apt/lists/* | |
| # copy files | |
| WORKDIR /app | |
| COPY . . | |
| # Install libraries | |
| RUN python3 -m pip install -r requirements.txt | |
| # start bash shell | |
| CMD bash | |