-
Notifications
You must be signed in to change notification settings - Fork 96
/
environment.yml
52 lines (47 loc) · 1.68 KB
/
environment.yml
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
# Use this environment file if your computer has a nvidia GPU and runs Windows or Linux.
name: sleap
channels:
- conda-forge
- nvidia
- sleap
- anaconda
dependencies:
# Packages SLEAP uses directly
- conda-forge::attrs >=21.2.0 #,<=21.4.0
- conda-forge::cattrs ==1.1.1
- conda-forge::imageio-ffmpeg # Required for imageio to read/write videos with ffmpeg
- conda-forge::jsmin
- conda-forge::jsonpickle ==1.2
- conda-forge::networkx
- anaconda::numpy >=1.19.5,<1.23.0
- conda-forge::opencv <4.9.0
- conda-forge::h5py <=3.7.0
- conda-forge::pandas
- conda-forge::pip
- conda-forge::pillow #>=8.3.1,<=8.4.0
- conda-forge::psutil
- conda-forge::pykalman
- conda-forge::pyside2 >=5.12 # To ensure application works correctly with QtPy.
- conda-forge::python ~=3.7 # Run into _MAX_WINDOWS_WORKERS not found if ==
- conda-forge::python-rapidjson
- conda-forge::pyyaml
- conda-forge::pyzmq
- conda-forge::qtpy >=2.0.1
- conda-forge::rich
- conda-forge::scipy >=1.4.1,<=1.9.0
- conda-forge::scikit-image
- conda-forge::scikit-learn ==1.0
- conda-forge::scikit-video
- conda-forge::seaborn
- sleap/label/dev::tensorflow ==2.7.0 # TODO: Switch to main label when updated
- conda-forge::tensorflow-hub # Pinned in meta.yml, but no problems here... yet
- conda-forge::qudida
- conda-forge::albumentations
- conda-forge::ndx-pose
# Packages required by tensorflow to find/use GPUs
- conda-forge::cudatoolkit ==11.3.1
# "==" results in package not found
- conda-forge::cudnn=8.2.1
- nvidia::cuda-nvcc=11.3
- pip:
- "--editable=.[conda_dev]"