Dockerfile.pytorch

# This dockerfile is meant to be personalized, and serves as a template and demonstration.
# Modify it directly, but it is recommended to copy this dockerfile into a new build context (directory),
# modify to taste and modify docker-compose.yml.template to build and run it.

# It is recommended to control docker containers through 'docker-compose' https://docs.docker.com/compose/
# Docker compose depends on a .yml file to control container sets
# rocm-setup.sh can generate a useful docker-compose .yml file
# `docker-compose run --rm <rocm-terminal>`

# If it is desired to run the container manually through the docker command-line, the following is an example
# 'docker run -it --rm -v [host/directory]:[container/directory]:ro <user-name>/<project-name>'.

FROM ubuntu:16.04
MAINTAINER Michael Wootton <michael.wootton@amd>

# Initialize the image
# Modify to pre-install dev tools and ROCm packages
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends curl && \
  curl -sL http://repo.radeon.com/rocm/apt/.apt_1.9.2/rocm.gpg.key | apt-key add - && \
  sh -c 'echo deb [arch=amd64] http://repo.radeon.com/rocm/apt/.apt_1.9.2 xenial main > /etc/apt/sources.list.d/rocm.list' && \
  apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
  sudo \
  libelf1 \
  build-essential \
  bzip2 \
  ca-certificates \
  cmake \
  ssh \
  apt-utils \
  pkg-config \
  g++-multilib \
  gdb \
  git \
  less \
  libunwind-dev \
  libfftw3-dev \
  libelf-dev \
  libncurses5-dev \
  libomp-dev \
  libpthread-stubs0-dev \
  make \
  miopen-hip \
  miopengemm \
#  python-dev \
#  python-yaml \
#  python-pip \
  vim \
  libssl-dev \
  libboost-dev \
  libboost-system-dev \
  libboost-filesystem-dev \
  libopenblas-dev \
  rpm \
  wget \
  net-tools \
  iputils-ping \
  libnuma-dev \
  rocm-dev \
  rocrand \
  rocblas \
  rocfft \
  hipsparse && \
#  hip-thrust && \
  curl -sL https://apt.llvm.org/llvm-snapshot.gpg.key | apt-key add - && \
  sh -c 'echo deb [arch=amd64] http://apt.llvm.org/xenial/ llvm-toolchain-xenial-7 main > /etc/apt/sources.list.d/llvm7.list' && \
  sh -c 'echo deb-src http://apt.llvm.org/xenial/ llvm-toolchain-xenial-7 main >> /etc/apt/sources.list.d/llvm7.list' && \
  apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
  clang-7 && \
  apt-get clean && \
  rm -rf /var/lib/apt/lists/*

# fix capitalization in some cmake files...
RUN sed -i 's/find_dependency(hip)/find_dependency(HIP)/g' /opt/rocm/rocsparse/lib/cmake/rocsparse/rocsparse-config.cmake
RUN sed -i 's/find_dependency(hip)/find_dependency(HIP)/g' /opt/rocm/rocfft/lib/cmake/rocfft/rocfft-config.cmake
RUN sed -i 's/find_dependency(hip)/find_dependency(HIP)/g' /opt/rocm/miopen/lib/cmake/miopen/miopen-config.cmake
RUN sed -i 's/find_dependency(hip)/find_dependency(HIP)/g' /opt/rocm/rocblas/lib/cmake/rocblas/rocblas-config.cmake

# Grant members of 'sudo' group passwordless privileges
# Comment out to require sudo
#COPY sudo-nopasswd /etc/sudoers.d/sudo-nopasswd

# This is meant to be used as an interactive developer container
# Create user rocm-user as member of sudo group
# Append /opt/rocm/bin to the system PATH variable
#RUN useradd --create-home -G sudo --shell /bin/bash rocm-user
#RUN usermod -a -G video rocm-user
#    sed --in-place=.rocm-backup 's|^\(PATH=.*\)"$|\1:/opt/rocm/bin"|' /etc/environment

#USER rocm-user
#WORKDIR /home/rocm-user
WORKDIR /root
ENV PATH="${PATH}:/opt/rocm/bin" HIP_PLATFORM="hcc"

# Fix rocm_agent_enumerator
RUN \
  wget https://raw.githubusercontent.com/RadeonOpenCompute/rocminfo/master/rocm_agent_enumerator &&\
  cp rocm_agent_enumerator /opt/rocm/bin/rocm_agent_enumerator 

# Install latest hip-thrust
RUN \
  git clone https://github.com/ROCmSoftwarePlatform/Thrust.git && \
  cd Thrust && \
  git submodule init && \
  git submodule update && \
  ln -s /root/Thrust /opt/rocm/Thrust && \
  ln -s /opt/rocm/Thrust/thrust /opt/rocm/include/thrust

RUN apt-get update && \
  DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends software-properties-common \
  libsm6 \
  libxext6 \
  libfontconfig1 \
  libxrender1 \
  cxlactivitylogger

# Install Python 3.6 as default python
RUN add-apt-repository -y ppa:deadsnakes/ppa
RUN apt-get update && \
  DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends python3.6 python3.6-dev
RUN \
  update-alternatives --install /usr/bin/python python /usr/bin/python3.5 1 && \
  update-alternatives --install /usr/bin/python python /usr/bin/python3.6 2 && \
  curl https://bootstrap.pypa.io/get-pip.py | python3.6
RUN \
  pip install pyyaml
RUN \
  pip install opencv-python
RUN \
  pip install imgaug
RUN \
  pip install cython
RUN \
  pip install bcolz
RUN \
  pip install graphviz
RUN \
  pip install isoweek
RUN \
  pip install tqdm

#RUN \
#  curl -O https://repo.continuum.io/archive/Anaconda3-5.2.0-Linux-x86_64.sh && \
#  bash Anaconda3-5.2.0-Linux-x86_64.sh -b && \
#  rm Anaconda3-5.2.0-Linux-x86_64.sh

# The following are optional enhancements for the command-line experience
# Uncomment the following to install a pre-configured vim environment based on http://vim.spf13.com/
# 1.  Sets up an enhanced command line dev environment within VIM
# 2.  Aliases GDB to enable TUI mode by default
#RUN curl -sL https://j.mp/spf13-vim3 | bash && \
#    echo "alias gdb='gdb --tui'\n" >> ~/.bashrc

#RUN \
#  bash installers/Anaconda3-5.2.0-Linux-x86_64.sh -b

#ENV PATH="/home/rocm-user/anaconda3/bin:${PATH}" KMTHINLTO="1"
ENV KMTHINLTO="1" LANG="C.UTF-8" LC_ALL="C.UTF-8"

RUN \
  pip install setuptools

#RUN \
#  pip install pyyaml

#RUN \
#  pip install numpy scipy

RUN \
  pip install typing

RUN \
  pip install enum34

RUN \
  pip install hypothesis

RUN \
  pip install librosa

RUN \
  pip install pandas

RUN \
  pip install seaborn

RUN \
  pip install sklearn_pandas

RUN \
  pip install pandas_summary

RUN \
  pip install ipython

RUN \
  pip install jupyter

RUN \
  echo "alias jupyter='jupyter notebook --allow-root --ip=0.0.0.0'" >> /root/.bashrc

RUN \
  update-alternatives --install /usr/bin/gcc gcc /usr/bin/clang-7 50 && \
  update-alternatives --install /usr/bin/g++ g++ /usr/bin/clang++-7 50

#RUN \
#  git clone https://github.com/pytorch/vision.git 
# NOTE: Have to perform the following after pytorch is built and installed
#  cd vision && \
#  python setup.py install

# Default to a login shell
CMD ["bash", "-l"]