13 lines
579 B
Docker
13 lines
579 B
Docker
FROM nvidia/cuda:9.0-base
|
|
|
|
RUN apt update && apt install -y wget unzip curl bzip2 git
|
|
RUN curl -LO http://repo.continuum.io/miniconda/Miniconda-latest-Linux-x86_64.sh
|
|
RUN bash Miniconda-latest-Linux-x86_64.sh -p /miniconda -b
|
|
RUN rm Miniconda-latest-Linux-x86_64.sh
|
|
ENV PATH=/miniconda/bin:${PATH}
|
|
RUN conda update -y conda
|
|
|
|
RUN conda install -y pytorch torchvision -c pytorch
|
|
RUN mkdir /workspace/ && cd /workspace/ && git clone https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix.git && cd pytorch-CycleGAN-and-pix2pix && pip install -r requirements.txt
|
|
|
|
WORKDIR /workspace |