File size: 990 Bytes
1abc64f
3be620b
1abc64f
 
 
 
 
3be620b
1abc64f
 
 
 
d3b5bbb
3be620b
 
 
 
 
 
 
 
6689a7c
 
71d18a1
ff33c79
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
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

CMD ["python", "app.py"]