diff --git a/environment.yml b/environment.yml index f387e722c3..7f4701bd91 100644 --- a/environment.yml +++ b/environment.yml @@ -4,11 +4,11 @@ channels: - defaults dependencies: - python>=3.9 - - pip=20.3 - - cudatoolkit=11.3 - - pytorch=1.11.0 - - torchvision=0.12.0 - - numpy=1.19.2 + - pip>=22.2 + - cudatoolkit + - pytorch + - torchvision + - numpy - pip: - albumentations==0.4.3 - opencv-python==4.5.5.64