mirrors
/
PyRetri
mirror of https://github.com/PyRetri/PyRetri.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

torch version fix

pull/46/head
thomasflittner 2024-12-25 13:07:21 +09:00
parent 0fa0326146
commit f610033269
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -192,7 +192,7 @@ if __name__ == '__main__':
],
license='Apache License 2.0',
install_requires=[
'numpy', 'torch==2.3.1', 'torchvision==0.18.1', 'scikit-learn', 'yacs', 'tqdm',
'numpy', 'torch>=2.2.2', 'torchvision==0.18.1', 'scikit-learn', 'yacs', 'tqdm',
],
packages=find_packages(),
zip_safe=False)