如何在 Google-Colaboratory 上安装自定义包

数据挖掘 Python 深度学习 强化学习 开放式健身房
2022-03-06 22:29:22

我正在尝试在 GOOGLE COLAB 上安装软件包,但遇到导入错误,我无法导入主模块“gym”的子模块。

在此处输入图像描述

我做了以下事情。

首先,我通过 git 命令克隆了 git hub 存储库(!git clone https://github.com/zoraiz-ali/gym.git

然后我使用 sys.path 添加目录

import sys
sys.path.append('/content/gym')

我的 setup.py 文件的代码如下

from setuptools import setup

setup(
     name="gym_robot",
     version="0.3",
     url="https://github.com/zoraiz-ali/gym.git",
     author="Zoraiz Ali",
     license="MIT",
     packages=["gym_robot", "gym_robot.envs", "gym_robot.envs.helper", ],
     include_package_data=True,
     install_requires=["gym", "numpy", "opencv-python", "pillow"]
)

该软件包在我的计算机上运行良好。所以我不确定为什么我不能在 Colaboratory 上使用它。

任何人都知道解决方案,请检查这个。

2个回答

我认为这个问题是由于您的模块与另一个包同名,这是gym强化学习环境的包。这个包会自动安装在 google colab 环境中,所以当你尝试gym_robot gym它导入时,假设你想从现有包而不是你自己的模块导入它。由于包没有这个功能/模块,它会给出一个导入错误。尝试稍微更改您自己的模块的名称,这样它就不会干扰现有的gym包。

在笔记本的开头包含一个 pip install 命令。在此处查看其中一个笔记本,例如:https ://github.com/arangodb/interactive_tutorials/blob/master/notebooks/Graph_Retail_DA_I.ipynb