From 3c02774849702e7313c443b02fb27e4318d95c35 Mon Sep 17 00:00:00 2001 From: Jonghwan Choi Date: Wed, 16 Jul 2025 17:39:48 +0900 Subject: [PATCH] Update DEEPLABCUT.yaml enforce to install pytorch with cuda --- conda-environments/DEEPLABCUT.yaml | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/conda-environments/DEEPLABCUT.yaml b/conda-environments/DEEPLABCUT.yaml index a467158da2..2f655168e4 100644 --- a/conda-environments/DEEPLABCUT.yaml +++ b/conda-environments/DEEPLABCUT.yaml @@ -25,6 +25,8 @@ channels: - defaults dependencies: - python=3.10 + - pytorch=2.5.1=cuda126* + - torchvision - pip - ipython - jupyter @@ -33,6 +35,6 @@ dependencies: - ffmpeg - pytables==3.8.0 - pip: - - torch - - torchvision +# - torch +# - torchvision - deeplabcut[gui,modelzoo,wandb]==3.0.0rc9