From ee9aa3dba3de5ec577213537544ade6cffdff4d3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cosmin=20=C8=98tefan=20Ciocan?=
 <57830279+cosminc98@users.noreply.github.com>
Date: Sat, 27 Jan 2024 14:45:16 +0000
Subject: [PATCH] Change devcontainer base image back to ubuntu:22.04 to match
 the environment from the test runner

---
 .devcontainer/Dockerfile | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile
index 0f46540..f5e11b2 100644
--- a/.devcontainer/Dockerfile
+++ b/.devcontainer/Dockerfile
@@ -1,17 +1,27 @@
-FROM nvcr.io/nvidia/cuda:12.3.1-devel-ubuntu22.04
+FROM ubuntu:22.04
 
 ARG VENV_PATH=/opt/dev-venv
 ENV VENV_ACTIVATE=${VENV_PATH}/bin/activate
 ENV DEBIAN_FRONTEND="noninteractive"
 
+# install the latest CUDA toolkit (https://developer.nvidia.com/cuda-downloads)
 RUN apt update
+RUN apt install -y wget
+RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
+RUN dpkg -i cuda-keyring_1.1-1_all.deb
+RUN apt update
+RUN apt -y install cuda-toolkit-12-3
+RUN echo "PATH=\"\$PATH:/usr/local/cuda/bin\"" >> ~/.bashrc
+
+# install OpenCV to test compilation with external libraries
+RUN apt install -y libopencv-dev pkg-config
+
+# make & language-pack-en are for documentation
 RUN apt install -y \
     gcc \
     git \
     language-pack-en \
-    libopencv-dev \
     make \
-    pkg-config \
     python3.10-venv \
     vim