-
Notifications
You must be signed in to change notification settings - Fork 0
/
conda_env.yml
35 lines (35 loc) · 858 Bytes
/
conda_env.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
name: btseg-env
channels:
- pytorch
- nvidia
- conda-forge
- defaults
dependencies:
- python=3.10.*
- python-devtools # for installing spatial-correlation-sampler
- pytorch::pytorch=2.0.1
- torchvision
- pytorch-cuda=11.7
- cudatoolkit # for installing
- cudatoolkit-dev # spatial-correlation-sampler
- jupyter
- notebook
- opencv
# - wandb # optional
- h5py
- pip
- pudb # debugger
- pip:
- largestinteriorrectangle
- pytorch-lightning[extra]==2.0.9
- torchmetrics
- jsonargparse[signatures]
- omegaconf
- h5py
- tensorboard
- pyrootutils # standardizing the project root setup
- rich
- seaborn
- matplotlib
- python-devtools # for installing spatial-correlation-sampler
- spatial-correlation-sampler # for computing the spatial correlation