diff options
Diffstat (limited to '.devcontainer')
-rw-r--r-- | .devcontainer/Dockerfile | 26 | ||||
-rw-r--r-- | .devcontainer/devcontainer.json | 4 |
2 files changed, 22 insertions, 8 deletions
diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile index 3df6b99..99fb890 100644 --- a/.devcontainer/Dockerfile +++ b/.devcontainer/Dockerfile @@ -8,16 +8,28 @@ 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} \ + cuda-nvcc-${CUDA_PKG_VERSION} ARG ROCM_VERSION=6.2.2 RUN mkdir --parents --mode=0755 /etc/apt/keyrings && \ @@ -29,9 +41,11 @@ RUN mkdir --parents --mode=0755 /etc/apt/keyrings && \ rocminfo \ rocm-gdb \ rocm-smi-lib \ - hip-runtime-amd && \ + rocm-llvm-dev \ + hip-runtime-amd && \ + hip-dev && \ 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 diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json index 7cae35a..34e88fb 100644 --- a/.devcontainer/devcontainer.json +++ b/.devcontainer/devcontainer.json @@ -28,7 +28,7 @@ //"hostRequirements": { "gpu": "optional" } "customizations": { "vscode": { - "extensions": [ "mhutchie.git-graph" ], + "extensions": [ "mhutchie.git-graph" ] } -} + } } |