FROM tensorflow/tensorflow:2.7.0 | |
# Because of https://developer.nvidia.com/blog/updating-the-cuda-linux-gpg-repository-key/ and https://github.com/NVIDIA/nvidia-docker/issues/1631#issuecomment-1112828208 | |
# RUN rm /etc/apt/sources.list.d/cuda.list | |
# RUN rm /etc/apt/sources.list.d/nvidia-ml.list | |
# RUN apt-key del 7fa2af80 | |
# RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub | |
# RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu2004/x86_64/7fa2af80.pub | |
# # Update and install ffmpeg | |
# RUN apt-get -y update | |
# RUN apt-get -y upgrade | |
# RUN apt-get install -y ffmpeg | |
RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y | |
# Setup environment | |
WORKDIR /GANime | |
ENV PROJECT_DIR=/GANime | |
COPY requirements.txt /GANime/requirements.txt | |
RUN pip install -r requirements.txt | |
COPY . . | |
RUN pip install -e . | |
RUN mkdir /.cache | |
RUN chmod 777 /.cache | |
RUN mkdir /GANime/flagged | |
RUN chmod 777 /GANime/flagged | |
CMD ["python", "app.py"] | |