From 626da1f1bd9cfa4be1823ef070dfc1c0e5aa99aa Mon Sep 17 00:00:00 2001 From: Henry Ruhs Date: Thu, 4 Apr 2024 16:48:00 +0200 Subject: [PATCH] Use modern cudnn --- install.js | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/install.js b/install.js index 325bf9d..1e14708 100644 --- a/install.js +++ b/install.js @@ -37,7 +37,7 @@ module.exports = async kernel => { message: [ - 'conda install -y --override-channels cudatoolkit=11.8.0 cudnn=8.4.1.50 -c conda-forge' + 'conda install cudatoolkit=11.8 cudnn=8.9.2.26 -y' ], conda: {