How to setup pytorch in google-cloud-ml

前端 未结 2 607
無奈伤痛
無奈伤痛 2021-01-25 10:10

I try to throw job with Pytorch code in google-cloud-ml. so I code the \"setup.py\" file. And add option \"install_requires\"

\"setup.py\"

<
2条回答
  •  时光说笑
    2021-01-25 11:00

    The actual error message is a bit buried, but it is this:

    'install_requires' must be a string or list of strings containing valid project/version requirement specifiers; Invalid requirement, parse error at "'://downl'"

    To use packages not hosted on PyPI, you need to use dependency_links (see this documentation). Something like this ought to work:

    from setuptools import find_packages
    from setuptools import setup
    
    REQUIRED_PACKAGES = ['torchvision']
    DEPENDENCY_LINKS = ['http://download.pytorch.org/whl/cpu/torch-0.3.0.post4-cp27-cp27mu-linux_x86_64.whl']
    
    setup(
        name='trainer',
        version='0.1',
        install_requires=REQUIRED_PACKAGES,
        dependency_links=DEPENDENCY_LINKS,
        packages=find_packages(),
        include_package_data=True,
        description='My keras trainer application package.'
    )
    

提交回复
热议问题