From 4b0c521ae05b9008c7acbad8e3d876c832a5dc6d Mon Sep 17 00:00:00 2001
From: Guilhem Saurel <guilhem.saurel@laas.fr>
Date: Fri, 12 Jan 2024 21:02:55 +0100
Subject: [PATCH] conda: fix pytorch packaging

---
 environment.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/environment.yml b/environment.yml
index 9f7c9d8f..6dde614e 100644
--- a/environment.yml
+++ b/environment.yml
@@ -6,10 +6,10 @@ channels:
    - anaconda
    - defaults
 dependencies:
-  - nvidia::cudatoolkit==12.1
+  - pytorch-cuda==12.1
   - python=3.9
   - pip
-  - pytorch::pytorch==2.1
+  - pytorch==2.1
   - torchvision==0.16
   - geckodriver
   - firefox