-
Notifications
You must be signed in to change notification settings - Fork 27
/
environment.yaml
37 lines (32 loc) · 1.19 KB
/
environment.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# reasons you might want to use `environment.yaml` instead of `requirements.txt`:
# - pip installs packages in a loop, without ensuring dependencies across all packages
# are fulfilled simultaneously, but conda achieves proper dependency control across
# all packages
# - conda allows for installing packages without requiring certain compilers or
# libraries to be available in the system, since it installs precompiled binaries
name: myenv
channels:
- conda-forge
- defaults
# it is strongly recommended to specify versions of packages installed through conda
# to avoid situation when version-unspecified packages install their latest major
# versions which can sometimes break things
# current approach below keeps the dependencies in the same major versions across all
# users, but allows for different minor and patch versions of packages where backwards
# compatibility is usually guaranteed
dependencies:
- cudatoolkit=11.8.*
- hydra-core=1.*
- rich=13.*
- pre-commit=3.*
- pytest=7.*
- pip>=23
- pip:
- nvidia-cudnn-cu11==8.6.0.163
- tensorflow==2.13.*
- hydra-optuna-sweeper
- hydra-colorlog
- rootutils
- PyDOE
- scipy
- matplotlib