modify pytorch version

This commit is contained in:
tlk-dsg 2021-12-14 16:03:04 +08:00
parent 7dcc4378cb
commit cf2b207cec
1 changed files with 4 additions and 3 deletions

View File

@ -1,20 +1,21 @@
from setuptools import setup, find_packages from setuptools import setup, find_packages
setup( setup(
name='deepke', # 打包后的包文件名 name='deepke', # 打包后的包文件名
version='0.2.93', #版本号 version='0.2.94', #版本号
keywords=["pip", "RE","NER","AE"], # 关键字 keywords=["pip", "RE","NER","AE"], # 关键字
description='DeepKE 是基于 Pytorch 的深度学习中文关系抽取处理套件。', # 说明 description='DeepKE 是基于 Pytorch 的深度学习中文关系抽取处理套件。', # 说明
long_description="client", #详细说明 long_description="client", #详细说明
license="MIT", # 许可 license="MIT", # 许可
url='https://github.com/zjunlp/deepke', url='https://github.com/zjunlp/deepke',
author='ZJUNLP', author='ZJUNLP',
author_email='xx2020@zju.edu.cn', author_email='zhangningyu@zju.edu.cn',
include_package_data=True, include_package_data=True,
platforms="any", platforms="any",
package_dir={"": "src"}, package_dir={"": "src"},
packages=find_packages("src"), packages=find_packages("src"),
install_requires=[ install_requires=[
'torch==1.5', 'torch>=1.5,<=1.10',
'hydra-core==1.0.6', 'hydra-core==1.0.6',
'tensorboard==2.4.1', 'tensorboard==2.4.1',
'matplotlib==3.4.1', 'matplotlib==3.4.1',