浏览代码

adding new Dockerfile for ml-agents

/develop/dockerfile
Anupam Bhatnagar 5 年前
当前提交
c09b8549
共有 1 个文件被更改,包括 29 次插入0 次删除
  1. 29
      ml-agents.Dockerfile

29
ml-agents.Dockerfile


FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04
LABEL maintainer=ml-agents@unity3d.com
RUN apt update && apt -y upgrade
# install relevant tools
RUN apt install -y wget curl vim tmux git gdebi-core build-essential python3-pip unzip
RUN python3 -m pip install --upgrade pip
RUN pip install setuptools==45.1.0
# install libxfont1, xvfb older version (used for environments with graphics)
RUN wget http://security.ubuntu.com/ubuntu/pool/main/libx/libxfont/libxfont1_1.5.1-1ubuntu0.16.04.4_amd64.deb
RUN wget http://security.ubuntu.com/ubuntu/pool/universe/x/xorg-server/xvfb_1.18.4-0ubuntu0.7_amd64.deb
RUN yes | gdebi libxfont1_1.5.1-1ubuntu0.16.04.4_amd64.deb
RUN yes | gdebi xvfb_1.18.4-0ubuntu0.7_amd64.deb
RUN apt-get install -y --no-install-recommends htop mesa-utils xorg-dev xorg libglvnd-dev libgl1-mesa-dev libegl1-mesa-dev libgles2-mesa-dev
ENV LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH
# install the latest version of ml-agents (from the master branch )
RUN git clone https://github.com/Unity-Technologies/ml-agents.git /ml-agents
WORKDIR /ml-agents
RUN git pull origin master
WORKDIR /ml-agents/ml-agents-envs
RUN pip install -e .
WORKDIR /ml-agents/ml-agents
RUN pip install -e .
# to build the docker image execute the following command in the folder containing this file
# docker build -t ml-agents:latest -f ml-agents.Dockerfile .
正在加载...
取消
保存