File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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
710setuptools >= 65.5.1
8- squarify >= 0.4.3
9- torch >= 1.12.1
10- torchvision >= 0.13.1
11+ zipp >= 3.15.0
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments