Skip to content

Commit

Permalink
Add CUDA executables to PATH in CUDA Docker images
Browse files Browse the repository at this point in the history
  • Loading branch information
jonatanklosko committed Aug 22, 2024
1 parent 75ca4a3 commit eca2dc1
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,8 @@ RUN distro="ubuntu$(. /etc/lsb-release; echo "$DISTRIB_RELEASE" | tr -d '.')" &&
apt-get install -y git cuda-nvcc-${CUDA_VERSION} cuda-libraries-${CUDA_VERSION} libcudnn9-cuda-$cuda_major && \
apt-get clean -y && rm -rf /var/lib/apt/lists/*

ENV PATH="/usr/local/nvidia/bin:/usr/local/cuda/bin:$PATH"

# Build stage: builds the release
FROM base-${VARIANT} AS build

Expand Down

0 comments on commit eca2dc1

Please sign in to comment.