Dockerfile example

Dockerfile that works for me if anyone’s interested.

FROM nvidia/cuda:11.0.3-devel-ubuntu20.04

ENV DEBIAN_FRONTEND=noninteractive

RUN apt-get update && \
    apt-get install python3 -y && \
    apt-get install python3-pip -y && \
	apt-get install git ffmpeg libsm6 libxext6 -y

RUN cd ./home && \
	git clone https://github.com/Janspiry/Palette-Image-to-Image-Diffusion-Models
#I only clone this for the requirements.txt.  I run palette from mounted drive /root/data

RUN pip3 install \
	torch==1.7.0+cu110 \
	torchvision==0.8.1+cu110 \ 
	-f https://download.pytorch.org/whl/torch_stable.html

RUN pip3 install -r ./home/Palette-Image-to-Image-Diffusion-Models/requirements.txt
	
WORKDIR ./root/data

Run with something like

docker run --rm --gpus all -it -v <path_to_palette_repo>:/root/data -v palette

[trufty]

I clone a copy to my local computer. So /root/data points to my local code path. That way I can edit the code on my local machine, and its available in the container via the mounted path.

Edit: I’m running docker with WSL as well

[Janspiry]

That’s good advice, but need some time. Considering that other users can work well, I think there is a bias in the operation.