From eca2dc1143930bfbe468ad6a453fda238b4cf9d5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jonatan=20K=C5=82osko?= <jonatanklosko@gmail.com>
Date: Thu, 22 Aug 2024 23:07:36 +0900
Subject: [PATCH] Add CUDA executables to PATH in CUDA Docker images

---
 Dockerfile | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 451432cf7..737143ca2 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -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