NASP/dockerfiles
35
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

try: use nvidia/cuda:11.3.1-devel-ubuntu20.04 as base image

Browse Source
This commit is contained in:
xiongdian.me 2023-04-29 19:18:56 +08:00
parent c8e6646009
commit b476a3e2cf
1 changed files with 1 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
nasp-ubuntu/Dockerfile
View File

@ -1,4 +1,4 @@
FROM docker.io/ubuntu:22.04 FROM docker.io/nvidia/cuda:11.3.1-devel-ubuntu20.04
# Install dependencies # Install dependencies
RUN ln -fs /usr/share/zoneinfo/Asia/Shanghai /etc/localtime \ RUN ln -fs /usr/share/zoneinfo/Asia/Shanghai /etc/localtime \
@ -17,16 +17,6 @@ RUN cd /tmp && wget http://192.168.16.118/MLNX_OFED_LINUX-5.8-2.0.3.0-ubuntu22.0
&& ./mlnxofedinstall --skip-unsupported-devices-check -q \ && ./mlnxofedinstall --skip-unsupported-devices-check -q \
&& rm -rf /tmp/MLNX_OFED_LINUX-5.8-* && rm -rf /tmp/MLNX_OFED_LINUX-5.8-*
# cuda
RUN apt-get update -y && apt-get install -yq linux-headers-$(uname -r) \
&& apt-key del 7fa2af80 && cd /tmp && wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb \
&& dpkg -i cuda-keyring_1.0-1_all.deb && apt-get update -y \
&& DEBIAN_FRONTEND=noninteractive apt-get install -yq --no-install-recommends cuda-11-3 cuda-compat-12-0
ENV PATH=/usr/local/cuda/bin:$PATH \
LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH \
CUDA_HOME=/usr/local/cuda
# startup # startup
COPY startup.sh /etc/startup.sh COPY startup.sh /etc/startup.sh
ENTRYPOINT ["/etc/startup.sh"] ENTRYPOINT ["/etc/startup.sh"]