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 # 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 ENV DEBIAN_FRONTEND=noninteractive
# packages # packages
RUN apt-get -y update && \ RUN apt-get -y update && \
apt-get install -y --no-install-recommends build-essential git nano rsync vim tree curl \ apt-get install -y --no-install-recommends build-essential git nano rsync vim tree curl wget \
wget unzip htop tmux xvfb patchelf ca-certificates bash-completion libjpeg-dev libpng-dev \ swig ffmpeg unzip htop tmux xvfb ca-certificates bash-completion libjpeg-dev libpng-dev \
ffmpeg cmake swig libssl-dev libcurl4-openssl-dev libopenmpi-dev python3-dev zlib1g-dev \ libssl-dev libcurl4-openssl-dev libopenmpi-dev zlib1g-dev qtbase5-dev qtdeclarative5-dev \
qtbase5-dev qtdeclarative5-dev libglib2.0-0 libglu1-mesa-dev libgl1-mesa-dev libvulkan1 \ libglib2.0-0 libglu1-mesa-dev libgl1-mesa-dev libvulkan1 libgl1-mesa-glx libosmesa6 \
libgl1-mesa-glx libosmesa6 libosmesa6-dev libglew-dev mesa-utils && \ libosmesa6-dev libglew-dev mesa-utils && \
apt-get clean && \ apt-get clean && \
apt-get autoremove -y && \ apt-get autoremove -y && \
rm -rf /var/lib/apt/lists/* && \ rm -rf /var/lib/apt/lists/* && \
mkdir /root/.ssh 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 # conda environment
COPY nvidia_icd.json /usr/share/vulkan/icd.d/nvidia_icd.json COPY nvidia_icd.json /usr/share/vulkan/icd.d/nvidia_icd.json
COPY environment.yaml /root COPY environment.yaml /root
@@ -48,7 +38,10 @@ RUN conda update conda && \
conda env update -n base -f /root/environment.yaml && \ conda env update -n base -f /root/environment.yaml && \
rm /root/environment.yaml && \ rm /root/environment.yaml && \
conda clean -ya && \ 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 # mujoco 2.1.0
ENV MUJOCO_GL egl ENV MUJOCO_GL egl

9
docker/environment.yaml
View File

@@ -9,9 +9,9 @@ dependencies:
- glib=2.78.4 - glib=2.78.4
- pip=24.0 - pip=24.0
- python=3.9 - python=3.9
- pytorch - pytorch=2.6.0.dev20241112
- pytorch-cuda=12.4 - pytorch-cuda=12.4
- torchvision - torchvision=0.15.2
- pip: - pip:
- absl-py==2.1.0 - absl-py==2.1.0
- "cython<3" - "cython<3"
@@ -29,9 +29,10 @@ dependencies:
- omegaconf==2.3.0 - omegaconf==2.3.0
- moviepy==1.0.3 - moviepy==1.0.3
- mujoco==2.3.1 - mujoco==2.3.1
- mujoco-py==2.1.2.14
- numpy==1.24.4 - numpy==1.24.4
- tensordict-nightly - tensordict-nightly==2024.11.14
- torchrl-nightly - torchrl-nightly==2024.11.14
- kornia==0.7.2 - kornia==0.7.2
- termcolor==2.4.0 - termcolor==2.4.0
- tqdm==4.66.4 - tqdm==4.66.4