Dockerfile 579 B

123456789101112131415161718192021222324
  1. FROM kaixhin/cuda-torch
  2. RUN apt-get update && apt-get install -y --no-install-recommends --force-yes \
  3. libsnappy-dev \
  4. graphicsmagick \
  5. libgraphicsmagick1-dev \
  6. libssl-dev \
  7. ca-certificates \
  8. git && \
  9. rm -rf /var/lib/apt/lists/*
  10. # https://github.com/nagadomi/waifu2x
  11. RUN \
  12. luarocks install graphicsmagick && \
  13. luarocks install lua-csnappy && \
  14. luarocks install md5 && \
  15. luarocks install uuid && \
  16. luarocks install csvigo && \
  17. PREFIX=$HOME/torch/install luarocks install turbo && \
  18. luarocks install cudnn
  19. COPY . /root/waifu2x
  20. WORKDIR /root/waifu2x