diff --git a/environment.yml b/environment.yml index 578a58a0..fb3e1ba3 100644 --- a/environment.yml +++ b/environment.yml @@ -10,7 +10,7 @@ dependencies: - pillow=8.3.1 - scipy=1.7.1 - pytorch=1.9.1 - - cudatoolkit=11.1 + - cudatoolkit>=11.1 - requests=2.26.0 - tqdm=4.62.2 - ninja=1.10.2