# Use the official image as a parent image. FROM nvidia/cuda:10.2-base CMD nvidia-smi # Set the working directory. WORKDIR /opt/ml/code/ # Copy the file from your host to your current location. COPY . . # Run the command inside your image filesystem. RUN apt-get update && \ apt-get install --no-install-recommends -y \ python3.6 python3-pip python3-setuptools python3-dev \ gcc swig libgl1-mesa-glx libglib2.0-0 && \ apt-get clean && rm -rf /var/lib/apt/lists/* RUN pip3 install --upgrade pip setuptools RUN pip3 install -r ./requirements.txt RUN pip3 install sagemaker-training # Run the specified command within the container. # CMD python ./train_climber.py # ENV TIMESTEPS 1e5 ENV SAGEMAKER_PROGRAM ./train_climber.py # CMD python SAGEMAKER_PROGRAM