在 google-cloud-ml 中，设置 PYTORCH 时出现 setup.py 错误

Question

我像下面这样编写 setup.py 来在 google-cloud-ml 引擎中设置 pytorch

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',
    dependency_links=DEPENDENCY_LINKS,
    install_requires=REQUIRED_PACKAGES,
    packages=find_packages(),
    include_package_data=True,
    description='My pytorch trainer application package.'
)

错误信息

"Command "python setup.py egg_info" failed with error code 1 in /tmp/pip-build-_ZQ7aQ/torch/"

我不知道会发生什么...... 当我搜索该问题时...答案只是升级 setuptools...但我不知道如何在 ml-engine 中升级 setuptools...

所以请帮助我...！ 我想在 ml 引擎中运行 pytorch 代码

Answer 1

似乎 DEPENDENCY_LINKS 已被 pip 忽略。 相反，我将 whl 文件复制到 GCS 存储桶，并在 gcloud 中使用标志 '--package gs://my-bucket/torch-0.3.0.post4-cp27-cp27mu-linux_x86_64.whl' 来安装 whl在执行 'pip install torchvision' 之前的文件并且它起作用了。

您还需要从 setup.py 中删除 DEPENDENCY_LINKS

from setuptools import find_packages
from setuptools import setup
REQUIRED_PACKAGES = ['torchvision']
setup(
    name='trainer',
    version='0.1',
    install_requires=REQUIRED_PACKAGES,
    packages=find_packages(),
    include_package_data=True,
    description='My pytorch trainer application package.'
)