add: adding docker file and requirements.txt #408

Original file line number	Diff line number	Diff line change
		@@ -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/ubu ntu1804/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/ubu ntu1804/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"]

15 changes: 15 additions & 0 deletions README.md

Show comments

Original file line number	Diff line number	Diff line change
Expand Up		@@ -188,6 +188,21 @@ For academic and non-commercial use only.The whole project is under the CC-BY-NC

		[![logo](./docs/img/vggface2_hq_compare.png)](https://github.com/NNNNAI/VGGFace2-HQ)

		## Installation

		```bash
		# Git clone
		git clone
		# Download weights
		bash download_weights.sh
		# Build Docker file
		docker build -t simswap .
		# Run docker file
		docker run -it -v :/simpswap --gpus=all -p 8888:8888 simswap:latest
		# Access jupyter notebook
		http://localhost:8888/tree/simpswap
		```

		Learn about our other projects

		[[VGGFace2-HQ]](https://github.com/NNNNAI/VGGFace2-HQ);
Expand Down

11 changes: 11 additions & 0 deletions requirements.txt

Show comments

Navigation Menu