From 875090eda3f4296eb0589817b70cb08f30b1b85b Mon Sep 17 00:00:00 2001
From: Hossein Rashidi <h.rashidi128@gmail.com>
Date: Mon, 28 Oct 2024 12:41:19 +0000
Subject: [PATCH] update cudnn version to 9

---
 .devcontainer/Dockerfile | 2 +-
 tests.dstack.yml         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile
index a02180f1..c9ee5d1e 100644
--- a/.devcontainer/Dockerfile
+++ b/.devcontainer/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
+FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04
 RUN apt-get update && apt-get install -y libgl1 libglib2.0-0 ffmpeg locales
 
 # Set the locale
diff --git a/tests.dstack.yml b/tests.dstack.yml
index 8dbd9ef3..8e43d8ee 100644
--- a/tests.dstack.yml
+++ b/tests.dstack.yml
@@ -4,7 +4,7 @@ name: aana-tests
 
 backends: [runpod]
 
-image: nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
+image: nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04
 
 env:
   - HF_TOKEN