diff options
Diffstat (limited to '.devcontainer/Dockerfile')
| -rw-r--r-- | .devcontainer/Dockerfile | 22 |
1 files changed, 17 insertions, 5 deletions
diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile index 3df6b99..f767657 100644 --- a/.devcontainer/Dockerfile +++ b/.devcontainer/Dockerfile @@ -8,16 +8,27 @@ RUN DEBIAN_FRONTEND=noninteractive apt-get update -y && DEBIAN_FRONTEND=noninter python3 \ ripgrep \ git \ - ltrace + ltrace \ + # required by llvm 17 + lsb-release software-properties-common gnupg + +ARG LLVM_VERSION=17 +RUN wget https://apt.llvm.org/llvm.sh && \ + chmod +x 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_VERSION=12-4 +ARG CUDA_PKG_VERSION=12-4 # Docker <-> host driver version compatiblity is newer host <-> older docker # We don't care about a specific driver version, so pick oldest 5XX ARG CUDA_DRIVER=515 -RUN 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 + libgl-dev libegl-dev libvdpau-dev \ nvidia-utils-${CUDA_DRIVER} \ - cuda-cudart-${CUDA_VERSION} + cuda-cudart-dev-${CUDA_PKG_VERSION} \ + cuda-cudart-${CUDA_PKG_VERSION} \ + cuda-profiler-api-${CUDA_PKG_VERSION} ARG ROCM_VERSION=6.2.2 RUN mkdir --parents --mode=0755 /etc/apt/keyrings && \ @@ -29,9 +40,10 @@ RUN mkdir --parents --mode=0755 /etc/apt/keyrings && \ rocminfo \ rocm-gdb \ rocm-smi-lib \ + rocm-llvm-dev \ hip-runtime-amd && \ echo '/opt/rocm/lib' > /etc/ld.so.conf.d/rocm.conf && \ ldconfig -ENV PATH=$PATH:/opt/rocm-6.2.2/bin +ENV PATH=$PATH:/opt/rocm-${ROCM_VERSION}/bin |