add: adding docker file and requirements.txt

This commit is contained in:
sohaibanwaar
2023-05-13 15:06:17 +05:00
parent 7a6c92ae69
commit ca730ada1b
3 changed files with 52 additions and 0 deletions
+26
View File
@@ -0,0 +1,26 @@
FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-devel
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub && \
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
RUN apt-get update && \
apt-get install -y wget && \
wget -qO - https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub | apt-key add - && \
apt-get update && \
apt-get install -y git && \
apt-get install ffmpeg libsm6 libxext6 -y && \
apt install -y libprotobuf-dev protobuf-compiler && \
apt-get clean
RUN pip install --upgrade pip
RUN pip install --upgrade setuptools
WORKDIR /simpswap
COPY ./requirements.txt /simpswap/requirements.txt
RUN pip install -r /simpswap/requirements.txt
# Use external volume for data
ENV NVIDIA_VISIBLE_DEVICES 1
CMD ["jupyter", "notebook", "--ip=0.0.0.0", "--port=8888", "--allow-root"]