diff --git a/environment.yml b/environment.yml index 00418ef8c..9d14ff5be 100644 --- a/environment.yml +++ b/environment.yml @@ -8,8 +8,8 @@ dependencies: - eigen==3.3.7 - protobuf==3.11.4 - libprotobuf==3.11.4 # We need to avoid problems with paths (idk why) - - cudnn==8.0 - - cudatoolkit==10.2 + - cudnn==8.0.5.39 + - cudatoolkit==10.2.89 - gtest - graphviz # Build & Run - wget