diff --git a/environment.yaml b/environment.yaml index b9abe569..3f157c60 100644 --- a/environment.yaml +++ b/environment.yaml @@ -1,10 +1,12 @@ name: midas-py310 channels: + - nvidia - pytorch + - conda-forge - defaults dependencies: - - nvidia::cudatoolkit=11.7 - python=3.10.8 + - cudatoolkit=11.7 - pytorch::pytorch=1.13.0 - torchvision=0.14.0 - pip=22.3.1 @@ -13,4 +15,4 @@ dependencies: - opencv-python==4.6.0.66 - imutils==0.5.4 - timm==0.6.12 - - einops==0.6.0 \ No newline at end of file + - einops==0.6.0