mirror of
https://github.com/vosen/ZLUDA.git
synced 2025-09-21 08:49:04 +00:00
Update devcontainer (#507)
This commit is contained in:
parent
e3a99b7ee1
commit
044fab47e5
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 \
|
||||
wget \
|
||||
|
@ -18,14 +18,16 @@ RUN wget https://apt.llvm.org/llvm.sh && \
|
|||
./llvm.sh ${LLVM_VERSION}
|
||||
|
||||
# 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
|
||||
# We don't care about a specific driver version, so pick oldest 5XX compatible
|
||||
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 && \
|
||||
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
|
||||
# It used to be possible to install cudnn8 and cudnn9 side-by-side, but now cudnn9 explicitly
|
||||
# conflicts with cudnn8, we will resolve it later if needed
|
||||
# RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/libcudnn8_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 \
|
||||
# CUDA headers need it for interop
|
||||
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-profiler-api-${CUDA_PKG_VERSION} \
|
||||
cuda-nvcc-${CUDA_PKG_VERSION} \
|
||||
libcudnn8-dev \
|
||||
# libcudnn8-dev \
|
||||
cudnn9-cuda-${CUDA_PKG_VERSION} \
|
||||
libcufft-dev-${CUDA_PKG_VERSION} \
|
||||
libcublas-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 && \
|
||||
wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | \
|
||||
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 \
|
||||
rocminfo \
|
||||
rocm-gdb \
|
||||
rocm-smi-lib \
|
||||
rocm-smi-lib amd-smi-lib \
|
||||
rocm-llvm-dev \
|
||||
hip-runtime-amd \
|
||||
miopen-hip-dev \
|
||||
rocfft-dev \
|
||||
rocblas-dev \
|
||||
hipblaslt-dev \
|
||||
rocsolver-dev \
|
||||
rocsparse-dev \
|
||||
hip-dev && \
|
||||
echo '/opt/rocm/lib' > /etc/ld.so.conf.d/rocm.conf && \
|
||||
ldconfig
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue