mirror of
https://github.com/livebook-dev/livebook.git
synced 2024-09-20 01:55:56 +08:00
Add CUDA executables to PATH in CUDA Docker images
This commit is contained in:
parent
75ca4a3734
commit
eca2dc1143
|
@ -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 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/*
|
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
|
# Build stage: builds the release
|
||||||
FROM base-${VARIANT} AS build
|
FROM base-${VARIANT} AS build
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue