hongshaorou/tdmpc2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update Dockerfile + environment.yaml

Browse Source
This commit is contained in:
Nicklas Hansen
2024-11-28 11:35:15 -08:00
parent 1bfbcb7794
commit 0a79c8bd38
2 changed files with 15 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
docker/Dockerfile
View File

@@ -16,31 +16,21 @@
##########################################
# base image
FROM nvidia/cudagl:11.3.1-devel-ubuntu20.04
FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime
ENV DEBIAN_FRONTEND=noninteractive
# packages
RUN apt-get -y update && \
apt-get install -y --no-install-recommends build-essential git nano rsync vim tree curl \
wget unzip htop tmux xvfb patchelf ca-certificates bash-completion libjpeg-dev libpng-dev \
ffmpeg cmake swig libssl-dev libcurl4-openssl-dev libopenmpi-dev python3-dev zlib1g-dev \
qtbase5-dev qtdeclarative5-dev libglib2.0-0 libglu1-mesa-dev libgl1-mesa-dev libvulkan1 \
libgl1-mesa-glx libosmesa6 libosmesa6-dev libglew-dev mesa-utils && \
apt-get install -y --no-install-recommends build-essential git nano rsync vim tree curl wget \
swig ffmpeg unzip htop tmux xvfb ca-certificates bash-completion libjpeg-dev libpng-dev \
libssl-dev libcurl4-openssl-dev libopenmpi-dev zlib1g-dev qtbase5-dev qtdeclarative5-dev \
libglib2.0-0 libglu1-mesa-dev libgl1-mesa-dev libvulkan1 libgl1-mesa-glx libosmesa6 \
libosmesa6-dev libglew-dev mesa-utils && \
apt-get clean && \
apt-get autoremove -y && \
rm -rf /var/lib/apt/lists/* && \
mkdir /root/.ssh
# miniconda
RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
/bin/bash ~/miniconda.sh -b -p /opt/conda && \
rm ~/miniconda.sh && \
. /opt/conda/etc/profile.d/conda.sh && \
conda init && \
conda clean -ya
ENV PATH /opt/conda/bin:$PATH
SHELL ["/bin/bash", "-c"]
# conda environment
COPY nvidia_icd.json /usr/share/vulkan/icd.d/nvidia_icd.json
COPY environment.yaml /root
@@ -48,7 +38,10 @@ RUN conda update conda && \
conda env update -n base -f /root/environment.yaml && \
rm /root/environment.yaml && \
conda clean -ya && \
pip cache purge
pip cache purge && \
conda init
SHELL ["/bin/bash", "-c"]
RUN echo "cd /root" >> /root/.bashrc
# mujoco 2.1.0
ENV MUJOCO_GL egl