Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Select an option

  • Save anjilinux/62f2ff37613e85c41400561e1aa84e16 to your computer and use it in GitHub Desktop.

Select an option

Save anjilinux/62f2ff37613e85c41400561e1aa84e16 to your computer and use it in GitHub Desktop.
Dockerfile - pytorch-ubuntu16.04 with only pip (without conda)
FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
RUN apt-get update && apt-get -y upgrade
RUN apt-get install -y --no-install-recommends git wget
RUN apt-get install -y software-properties-common vim ssh
RUN apt-get install -y python3
RUN apt-get install -y python3-dev python3-pip build-essential
RUN apt-get install -y software-properties-common vim ssh
RUN pip3 install --upgrade pip
RUN ln -s /usr/bin/python3 /usr/bin/python
RUN pip install torch torchvision
WORKDIR /workspace
ADD . .
RUN chmod -R a+w /workspace
EXPOSE 8000
EXPOSE 7822
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment