livebook-dev/livebook
1
1
Fork 0
mirror of https://github.com/livebook-dev/livebook.git synced 2024-09-20 01:55:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add CUDA executables to PATH in CUDA Docker images

Browse source
This commit is contained in:
Jonatan Kłosko 2024-08-22 23:07:36 +09:00
parent 75ca4a3734
commit eca2dc1143
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
Dockerfile
View file

@ -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