From ce0bee55e150ea0180fda152ff5f5698bed64fe7 Mon Sep 17 00:00:00 2001
From: Fortune Iriaye <firiaye@uab.edu>
Date: Thu, 20 Feb 2025 13:39:35 -0600
Subject: [PATCH] Update Dockerfile

---
 docker/Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 3f852ba..978613a 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -4,7 +4,8 @@ FROM tensorflow/tensorflow:1.15.0-gpu-py3
 # Set environment variables
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
 ENV DEBIAN_FRONTEND=noninteractive
-ENV PATH /opt/conda/bin:$PATH
+ENV PATH=/opt/conda/bin:$PATH
+ARG CUDA=12.2.2
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
-- 
GitLab