From 1723014e106dd8bbecd1063e32114e84e8151d03 Mon Sep 17 00:00:00 2001 From: Henry Ruhs Date: Mon, 18 Mar 2024 09:32:32 +0100 Subject: [PATCH] Fix cuda and cudnn values --- install.js | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/install.js b/install.js index 7797b56..602e35e 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 -y --override-channels cudatoolkit=11.8.0 cudnn=8.4.1.50 -c conda-forge' ], conda: {