diff --git a/setup.sh b/setup.sh
index 12bc849..a2271d3 100755
--- a/setup.sh
+++ b/setup.sh
@@ -1,5 +1,5 @@
 #!/usr/bin/env -S bash -i
 echo "Creating conda environment"
-mamba create -n 08_knowledge_extraction python=3.11 pytorch torchvision pytorch-cuda-12.1 -c conda-forge -c pytorch -c nvidia
+mamba create -n 08_knowledge_extraction python=3.11 pytorch torchvision pytorch-cuda=12.1 -c pytorch -c nvidia
 mamba activate 08_knowledge_extraction
 pip install -r requirements.txt
\ No newline at end of file