From 9371dde5436321fb87ca28b5d04de87140eb71f7 Mon Sep 17 00:00:00 2001 From: Francois Caud Date: Fri, 15 Dec 2023 15:08:06 +0100 Subject: [PATCH] pytorch-cuda --- environment.yml | 4 ++-- extra_libraries.txt | 3 +-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/environment.yml b/environment.yml index f040cf4..d520d49 100644 --- a/environment.yml +++ b/environment.yml @@ -1,6 +1,7 @@ name: ramp-stroke channels: - pytorch + - nvidia - conda-forge dependencies: - numpy==1.23.4 @@ -11,10 +12,9 @@ dependencies: - matplotlib-base # Extra requirements - - cudatoolkit=10.2 - pytorch - torchvision - - keras + - pytorch-cuda=11.8 - lightgbm - nilearn - scikit-image>=0.18 diff --git a/extra_libraries.txt b/extra_libraries.txt index 74ea467..c18d77b 100644 --- a/extra_libraries.txt +++ b/extra_libraries.txt @@ -1,7 +1,6 @@ imbalanced-learn scipy -torch # pytorch cudatoolkit=10.2 -keras +torch # pytorch pytorch-cuda=11.8 lightgbm xgboost catboost