mirror of
https://github.com/vosen/ZLUDA.git
synced 2025-09-28 20:29:11 +00:00
Update devcontainer
This commit is contained in:
parent
e3a99b7ee1
commit
e85efa097d
1 changed files with 17 additions and 9 deletions
|
@ -1,4 +1,4 @@
|
||||||
FROM nvidia/cuda:12.8.1-base-ubuntu24.04
|
FROM nvidia/cuda:12.9.1-base-ubuntu24.04
|
||||||
|
|
||||||
RUN DEBIAN_FRONTEND=noninteractive apt-get update -y && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
RUN DEBIAN_FRONTEND=noninteractive apt-get update -y && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
||||||
wget \
|
wget \
|
||||||
|
@ -18,14 +18,16 @@ RUN wget https://apt.llvm.org/llvm.sh && \
|
||||||
./llvm.sh ${LLVM_VERSION}
|
./llvm.sh ${LLVM_VERSION}
|
||||||
|
|
||||||
# Feel free to change to a newer version if you have a newer verison on your host
|
# Feel free to change to a newer version if you have a newer verison on your host
|
||||||
ARG CUDA_PKG_VERSION=12-8
|
ARG CUDA_PKG_VERSION=12-9
|
||||||
# Docker <-> host driver version compatiblity is newer host <-> older docker
|
# Docker <-> host driver version compatiblity is newer host <-> older docker
|
||||||
# We don't care about a specific driver version, so pick oldest 5XX compatible
|
# We don't care about a specific driver version, so pick oldest 5XX compatible
|
||||||
ARG CUDA_DRIVER=570
|
ARG CUDA_DRIVER=570
|
||||||
RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/libcudnn8_8.9.7.29-1+cuda12.2_amd64.deb && \
|
# It used to be possible to install cudnn8 and cudnn9 side-by-side, but now cudnn9 explicitly
|
||||||
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/libcudnn8-dev_8.9.7.29-1+cuda12.2_amd64.deb && \
|
# conflicts with cudnn8, we will resolve it later if needed
|
||||||
dpkg -i libcudnn8_8.9.7.29-1+cuda12.2_amd64.deb libcudnn8-dev_8.9.7.29-1+cuda12.2_amd64.deb && \
|
# RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/libcudnn8_8.9.7.29-1+cuda12.2_amd64.deb && \
|
||||||
rm libcudnn8_8.9.7.29-1+cuda12.2_amd64.deb libcudnn8-dev_8.9.7.29-1+cuda12.2_amd64.deb
|
# wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/libcudnn8-dev_8.9.7.29-1+cuda12.2_amd64.deb && \
|
||||||
|
# dpkg -i libcudnn8_8.9.7.29-1+cuda12.2_amd64.deb libcudnn8-dev_8.9.7.29-1+cuda12.2_amd64.deb && \
|
||||||
|
# rm libcudnn8_8.9.7.29-1+cuda12.2_amd64.deb libcudnn8-dev_8.9.7.29-1+cuda12.2_amd64.deb
|
||||||
RUN DEBIAN_FRONTEND=noninteractive apt-get update -y && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
RUN DEBIAN_FRONTEND=noninteractive apt-get update -y && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
||||||
# CUDA headers need it for interop
|
# CUDA headers need it for interop
|
||||||
libgl-dev libegl-dev libvdpau-dev \
|
libgl-dev libegl-dev libvdpau-dev \
|
||||||
|
@ -35,13 +37,13 @@ RUN DEBIAN_FRONTEND=noninteractive apt-get update -y && DEBIAN_FRONTEND=noninter
|
||||||
cuda-cudart-${CUDA_PKG_VERSION} \
|
cuda-cudart-${CUDA_PKG_VERSION} \
|
||||||
cuda-profiler-api-${CUDA_PKG_VERSION} \
|
cuda-profiler-api-${CUDA_PKG_VERSION} \
|
||||||
cuda-nvcc-${CUDA_PKG_VERSION} \
|
cuda-nvcc-${CUDA_PKG_VERSION} \
|
||||||
libcudnn8-dev \
|
# libcudnn8-dev \
|
||||||
cudnn9-cuda-${CUDA_PKG_VERSION} \
|
cudnn9-cuda-${CUDA_PKG_VERSION} \
|
||||||
libcufft-dev-${CUDA_PKG_VERSION} \
|
libcufft-dev-${CUDA_PKG_VERSION} \
|
||||||
libcublas-dev-${CUDA_PKG_VERSION} \
|
libcublas-dev-${CUDA_PKG_VERSION} \
|
||||||
libcusparse-dev-${CUDA_PKG_VERSION}
|
libcusparse-dev-${CUDA_PKG_VERSION}
|
||||||
|
|
||||||
ARG ROCM_VERSION=6.4
|
ARG ROCM_VERSION=6.4.3
|
||||||
RUN mkdir --parents --mode=0755 /etc/apt/keyrings && \
|
RUN mkdir --parents --mode=0755 /etc/apt/keyrings && \
|
||||||
wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | \
|
wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | \
|
||||||
gpg --dearmor | tee /etc/apt/keyrings/rocm.gpg > /dev/null && \
|
gpg --dearmor | tee /etc/apt/keyrings/rocm.gpg > /dev/null && \
|
||||||
|
@ -50,9 +52,15 @@ RUN mkdir --parents --mode=0755 /etc/apt/keyrings && \
|
||||||
DEBIAN_FRONTEND=noninteractive apt update -y && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
DEBIAN_FRONTEND=noninteractive apt update -y && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
||||||
rocminfo \
|
rocminfo \
|
||||||
rocm-gdb \
|
rocm-gdb \
|
||||||
rocm-smi-lib \
|
rocm-smi-lib amd-smi-lib \
|
||||||
rocm-llvm-dev \
|
rocm-llvm-dev \
|
||||||
hip-runtime-amd \
|
hip-runtime-amd \
|
||||||
|
miopen-hip-dev \
|
||||||
|
rocfft-dev \
|
||||||
|
rocblas-dev \
|
||||||
|
hipblaslt-dev \
|
||||||
|
rocsolver-dev \
|
||||||
|
rocsparse-dev \
|
||||||
hip-dev && \
|
hip-dev && \
|
||||||
echo '/opt/rocm/lib' > /etc/ld.so.conf.d/rocm.conf && \
|
echo '/opt/rocm/lib' > /etc/ld.so.conf.d/rocm.conf && \
|
||||||
ldconfig
|
ldconfig
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue