Skip to content

Commit

Permalink
Update environment configuration
Browse files Browse the repository at this point in the history
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)).
  • Loading branch information
aywi committed Nov 17, 2023
1 parent deeb3c3 commit 696ec03
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
8 changes: 4 additions & 4 deletions env_config/environment.yml
@@ -1,20 +1,20 @@
channels:
- nvidia
- pytorch
- nvidia
- defaults
dependencies:
- jupyter=1.0.0
- matplotlib=3.8.0
- numpy=1.26.0
- numpy=1.24.3
- pip=23.3
- python=3.11.5
- pytorch=2.1.1
- pytorch-cuda=11.8
- scikit-learn=1.2.2
- scikit-learn=1.3.0
- scipy=1.11.3
- torchaudio=2.1.1
- torchvision=0.16.1
- pip:
- captum==0.6.0
- opencv-contrib-python==4.8.1.78
- opencv-contrib-python-headless==4.8.1.78
- torchio==0.19.2
6 changes: 3 additions & 3 deletions env_config/environment_latest.yml
@@ -1,6 +1,6 @@
channels:
- nvidia
- pytorch
- nvidia
- defaults
dependencies:
- jupyter
Expand All @@ -10,11 +10,11 @@ dependencies:
- python=3.11
- pytorch
- pytorch-cuda=11
- scikit-learn
- scikit-learn>=1.3
- scipy
- torchaudio
- torchvision
- pip:
- captum
- opencv-contrib-python
- opencv-contrib-python-headless
- torchio

0 comments on commit 696ec03

Please sign in to comment.