From 6cce1cea306d5f2acac6b8f4a28c8d0d930ac8d6 Mon Sep 17 00:00:00 2001
From: NikoOinonen <nikooin@gmail.com>
Date: Tue, 23 Apr 2024 11:16:02 +0300
Subject: [PATCH] Update CUDA version.

The older CUDA version for some reason fails to compile the extensions on current Pytorch version.
---
 environment.yml | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/environment.yml b/environment.yml
index c05daf0..cf658b3 100644
--- a/environment.yml
+++ b/environment.yml
@@ -1,15 +1,14 @@
 name: ml-spm
 channels:
   - pytorch
-  - nvidia/label/cuda-11.8.0
+  - nvidia/label/cuda-12.1.0
   - conda-forge
   - defaults
 dependencies:
   - python=3.11
   - pip
   - pytorch
-  - torchvision
-  - torchaudio
+  - pytorch-cuda=12.1
   - cuda
   - ninja
   - matplotlib