Skip to content

Commit 696ec03

Browse files
committed
Update environment configuration
1. Force to use scikit-learn>=1.3 to avoid `DeprecationWarning` from NumPy. 2. Use opencv-contrib-python-headless to avoid conflicts with PyQt installed with conda (https://stackoverflow.com/q/46449850 opencv/opencv-python#46 (comment)).
1 parent deeb3c3 commit 696ec03

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

env_config/environment.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,20 @@
11
channels:
2-
- nvidia
32
- pytorch
3+
- nvidia
44
- defaults
55
dependencies:
66
- jupyter=1.0.0
77
- matplotlib=3.8.0
8-
- numpy=1.26.0
8+
- numpy=1.24.3
99
- pip=23.3
1010
- python=3.11.5
1111
- pytorch=2.1.1
1212
- pytorch-cuda=11.8
13-
- scikit-learn=1.2.2
13+
- scikit-learn=1.3.0
1414
- scipy=1.11.3
1515
- torchaudio=2.1.1
1616
- torchvision=0.16.1
1717
- pip:
1818
- captum==0.6.0
19-
- opencv-contrib-python==4.8.1.78
19+
- opencv-contrib-python-headless==4.8.1.78
2020
- torchio==0.19.2

env_config/environment_latest.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
channels:
2-
- nvidia
32
- pytorch
3+
- nvidia
44
- defaults
55
dependencies:
66
- jupyter
@@ -10,11 +10,11 @@ dependencies:
1010
- python=3.11
1111
- pytorch
1212
- pytorch-cuda=11
13-
- scikit-learn
13+
- scikit-learn>=1.3
1414
- scipy
1515
- torchaudio
1616
- torchvision
1717
- pip:
1818
- captum
19-
- opencv-contrib-python
19+
- opencv-contrib-python-headless
2020
- torchio

0 commit comments

Comments
 (0)