Skip to content

Commit 77c8a1d

Browse files
author
J. Cappelletto
committed
Update requirements and setup process to adopt Pytorch 2.0, also upgrading most of the core libraries and refrainig from using setup.py install step (only build)
1 parent c5b19f2 commit 77c8a1d

2 files changed

Lines changed: 20 additions & 18 deletions

File tree

environment/requirements.txt

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
1-
importlib_metadata>=5.0.0
2-
numpy>=1.23.4
3-
pandas>=1.4.3
4-
Pillow>=9.3.0
5-
scikit_learn>=1.1.3
6-
seaborn>=0.12.1
1+
# python == 3.10
2+
pandas == 1.5.3
3+
numpy == 1.24.2
4+
torch == 2.0.0
5+
torchvision == 0.15.1
6+
scikit-learn == 1.2.2
7+
Pillow == 9.4.0
8+
matplotlib == 3.7.1
9+
seaborn == 0.12.2
710
setuptools>=65.5.1
8-
squarify>=0.4.3
9-
torch>=1.12.1
10-
torchvision>=0.13.1
11+
zipp >= 3.15.0

setup.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -46,17 +46,18 @@ def run_setup():
4646
],
4747
},
4848

49-
# TODO: need to recreate dep matrix.
50-
# Pytorch does not support CUDA 11.5, using older 10.2 for the conda env
49+
# TODO: Update dependency matrix for pytorch=2.0
50+
# Pytorch now supports more 11.X series of CUDA
5151
install_requires = [
5252
"blitz-bayesian-pytorch==0.2.7",
53-
"numpy>=1.19.0", # check for update of pandas
54-
"pandas>=0.25.3",
55-
"torch>=1.7.0", # check for a newer version of torch, supporting multi-GPU and improved queries on allocated memory
56-
"torchvision>=0.8.1",
57-
"scikit-learn>=0.23.1", # scikit was bumped to 1.XX series. Maybe worth doing the update with the rest of the packages
58-
"Pillow>=9.1.1",
59-
"scipy>=1.5.0",
53+
# "numpy>=1.19.0", # check for update of pandas
54+
# "pandas>=0.25.3",
55+
# "torch>=1.7.0", # check for a newer version of torch, supporting multi-GPU and improved queries on allocated memory
56+
# "torchvision>=0.8.1",
57+
# "scikit-learn>=0.23.1", # scikit was bumped to 1.XX series. Maybe worth doing the update with the rest of the packages
58+
# "Pillow>=9.1.1",
59+
# "scipy>=1.5.0",
60+
# The rest of the dependencies are installed by $ pip -r requirements.txt
6061
]
6162
)
6263

0 commit comments

Comments
 (0)