|
|
@@ -1,4 +1,4 @@
|
|
|
-FROM nvidia/cuda:11.4.3-devel-ubuntu18.04
|
|
|
+FROM nvidia/cuda:11.4.3-cudnn8-devel-ubuntu18.04
|
|
|
|
|
|
ARG ROS_DISTRO=melodic
|
|
|
|
|
|
@@ -11,17 +11,38 @@ RUN usermod -aG sudo ${user}
|
|
|
RUN echo "${user}:888888" | chpasswd
|
|
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
+ENV NVIDIA_VISIBLE_DEVICES all
|
|
|
+ENV NVIDIA_DRIVER_CAPABILITIES compute,utility,graphics,display
|
|
|
+
|
|
|
# [Optional] Add sudo support. Omit if you don't need to install software after connecting.
|
|
|
RUN sed -i 's/http:\/\/archive.ubuntu.com\//http:\/\/mirrors.aliyun.com\//g' /etc/apt/sources.list \
|
|
|
&& apt-get update && apt-get install -y --no-install-recommends sudo \
|
|
|
- libc-bin \
|
|
|
- binutils \
|
|
|
+ curl \
|
|
|
+ gnupg2 \
|
|
|
lsb-release \
|
|
|
- curl
|
|
|
+ libgl1-mesa-dev \
|
|
|
+ libgl1-mesa-glx \
|
|
|
+ libglew-dev \
|
|
|
+ libosmesa6-dev \
|
|
|
+ software-properties-common \
|
|
|
+ libglvnd0 \
|
|
|
+ libgl1 \
|
|
|
+ libglx0 \
|
|
|
+ libegl1 \
|
|
|
+ libxext6 \
|
|
|
+ libx11-6 \
|
|
|
+ && rm -rf /var/lib/apt/lists/*
|
|
|
|
|
|
RUN sudo sh -c 'echo "deb http://packages.ros.org/ros/ubuntu $(lsb_release -sc) main" > /etc/apt/sources.list.d/ros-latest.list' \
|
|
|
&& curl -s https://raw.githubusercontent.com/ros/rosdistro/master/ros.asc | sudo apt-key add - \
|
|
|
- && apt-get update && apt-get install -y --no-install-recommends ros-melodic-desktop-full
|
|
|
+ && apt-get update && apt-get install -y \
|
|
|
+ ros-melodic-desktop-full \
|
|
|
+ python-rosdep \
|
|
|
+ python-rosinstall \
|
|
|
+ python-rosinstall-generator \
|
|
|
+ python-wstool \
|
|
|
+ build-essential \
|
|
|
+ && rm -rf /var/lib/apt/lists/*
|
|
|
|
|
|
# ********************************************************
|
|
|
# * Anything else you want to do like clean up goes here *
|
