Commit 77dc0a4e authored by Asa Rentschler's avatar Asa Rentschler
Browse files

Updated CUDA libs and a few other tweaks

parent c45f1b10
Loading
Loading
Loading
Loading
+9 −10
Original line number Diff line number Diff line
FROM quay.io/centos/centos:stream8
#FROM quay.io/nvidia/cuda-ppc64le:11.0.3-cudnn8-devel-centos8

# updated cuda libs to 12.2
RUN dnf -y install epel-release && dnf -y install fakeroot 
RUN dnf clean all && rm -r /var/cache/dnf  && fakeroot dnf upgrade -y && fakeroot dnf update -y  
RUN fakeroot dnf install -y wget dnf-plugins-core \
	&& fakeroot dnf config-manager --set-enabled powertools \
	&& wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda-repo-rhel8-11-0-local-11.0.3_450.51.06-1.ppc64le.rpm \
	&& fakeroot rpm -i cuda-repo-rhel8-11-0-local-11.0.3_450.51.06-1.ppc64le.rpm \
	&& wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda-repo-rhel8-12-2-local-12.2.0_535.54.03-1.ppc64le.rpm \
	&& fakeroot rpm -i cuda-repo-rhel8-12-2-local-12.2.0_535.54.03-1.ppc64le.rpm \
	&& fakeroot dnf -y module install nvidia-driver:latest-dkms \
	&& fakeroot dnf -y install cuda \
	&& fakeroot dnf -y install  kernel-devel \
		cuda-command-line-tools-11-0 \
		cuda-command-line-tools-12-2 \
		java-11-openjdk-devel \
		zip unzip \
		python3-devel \
		tzdata \
		gpg \
		glibc-devel \
	&& rm cuda-repo-rhel8-11-0-local-11.0.3_450.51.06-1.ppc64le.rpm \
	&& rm cuda-repo-rhel8-12-2-local-12.2.0_535.54.03-1.ppc64le.rpm \
	&& dnf -y clean all && rm -r /var/cache/dnf

COPY libcudnn8-8.2.0.53-1.cuda11.3.ppc64le.rpm /
COPY libcudnn8-devel-8.2.0.53-1.cuda11.3.ppc64le.rpm  /
RUN rpm -i /libcudnn8-8.2.0.53-1.cuda11.3.ppc64le.rpm \
	&& rpm -i /libcudnn8-devel-8.2.0.53-1.cuda11.3.ppc64le.rpm \
	&& rm /libcudnn8-8.2.0.53-1.cuda11.3.ppc64le.rpm /libcudnn8-devel-8.2.0.53-1.cuda11.3.ppc64le.rpm
# replaced old rpm installation with repo
RUN fakeroot dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/ppc64le/cuda-rhel8.repo
RUN fakeroot dnf -y install cuda-nvcc-12-2.ppc64le libcudnn8-8.9.7.29-1.cuda12.2.ppc64le libcudnn8-devel-8.9.7.29-1.cuda12.2.ppc64le && dnf clean all && rm -r /var/cache/dnf

#	&& dnf -y install tar wget git openssh 

ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/compat:/usr/local/cuda/lib64/stubs:$LD_LIBRARY_PATH
ENV PATH=/usr/local/cuda-11.0/bin:$PATH
ENV PATH=/usr/local/cuda-12.2/bin:$PATH
ENV CUDA_HOME="/usr/local/cuda"
ENV CUDA_ROOT="/usr/local/cuda"
ENV CPATH="/usr/local/cuda/include:$CPATH"
+1 −1
Original line number Diff line number Diff line
This image needs to be built somewhere where you have sudo or subuid/subgid mappings present

Currently built with CUDA 11.0.3 libraries
Currently built with CUDA 12.2 libraries
+1 −1
Original line number Diff line number Diff line
#!/bin/bash

podman build -f Dockerfile -t code.ornl.gov:4567/olcfcontainers/olcfbaseimages/baseimage-centos-cuda:latest .;
podman build --net=host -f Dockerfile -t code.ornl.gov:4567/olcfcontainers/olcfbaseimages/baseimage-centos-cuda:latest .;
#podman push code.ornl.gov:4567/olcfcontainers/olcfbaseimages/baseimage-centos-cuda:latest;
#podman save baseimage-centos-cuda.tar code.ornl.gov:4567/olcfcontainers/olcfbaseimages/baseimage-centos-cuda:latest;
#singularity build baseimage-centos-cuda.sif docker-archive://baseimage-centos-cuda.tar;
−484 MiB

File deleted.

−386 MiB

File deleted.

Loading