-
Notifications
You must be signed in to change notification settings - Fork 0
/
environment.yml
27 lines (27 loc) · 765 Bytes
/
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
name: ecl
channels:
- conda-forge
- pytorch
- nvidia
dependencies:
- python=3.10
- pip
- numpy
- pillow
- pandas
- tensorly
- wandb
- scikit-learn
- matplotlib
- pytorch
- torchvision
- pytorch-cuda=11.6
- pip:
- torchshow
# The released version of pretrainedmodels has a bug around DataParallel and
# pytorch 1.0, see the following links for details:
# - https://github.com/Cadene/pretrained-models.pytorch/pull/145
# - https://github.com/Cadene/pretrained-models.pytorch/issues/112
# - https://github.com/Cadene/pretrained-models.pytorch/issues/120
# - https://github.com/pytorch/pytorch/issues/8637
- "git+https://github.com/wpwei/pretrained-models.pytorch.git@vision_bug_fix#egg=pretrainedmodels"