安装说明 =============================== .. toctree:: :maxdepth: 2 前置条件 -------------- 系统版本: Linux, macOS, Windows Python 版本: 3.7-3.9 .. note:: 如果您的设备具有 NVIDIA GPU ,参考 `NVIDIA CUDA Toolkit 安装 `_。 在安装好 CUDA 之后,当您在安装 DI-engine 的依赖项时,会自动获取和安装带有 NVIDIA CUDA 加速的深度学习框架(例如PyTorch)。 如您需要手动安装适合本机 Python 版本和 GPU 设备的 PyTorch 版本,您可以参考 `PyTorch 安装 `_ 。 此外,如果您的操作系统是 Windows ,请您检查是否拥有 SWIG Windows 应用程序,您可以参考 `SWIG 安装 `_ ,并将 SWIG 的可执行文件添加至 Windows 系统环境变量 PATH 路径中。 发布版本 -------------- 您可以使用以下命令安装 DI-engine 的稳定发行版: .. code-block:: bash # Current stable release of DI-engine pip install DI-engine .. tip:: 如果你需要升级 pip 版本,可以使用以下命令: .. code-block:: bash # Windows > python -m pip install --upgrade pip # Linux $ pip install --upgrade pip # MacOS $ pip install --upgrade pip .. tip:: 如果您在下载软件包时遇到超时错误,您可以尝试采用更换其他常用镜像源,例如使用阿里源: .. code-block:: bash pip install requests -i https://mirrors.aliyun.com/pypi/simple/ DI-engine 如果您需要借助 Anaconda 或者 Miniconda , 建议您采用如下命令: .. code-block:: bash conda install -c opendilab di-engine 开发版本 -------------- 如果您需要从 Github 源码安装最新的 DI-engine 开发版本,可以使用如下方法: .. code-block:: bash git clone https://github.com/opendilab/DI-engine.git cd DI-engine pip install . .. tip:: 如果您希望将 DI-engine 的安装目录设置为操作系统的当前用户目录,您可以使用如下方式: .. code-block:: bash pip install . --user 如果您正在使用诸如 virtualenv 等程序生成的虚拟 python 环境,``--user`` 选项将不会生效。 特殊版本 -------------- 如果您希望启用 DI-engine 的额外功能并安装相关依赖,可以使用如下方法: .. code-block:: bash # install atari and box-2d related packages pip install DI-engine[common_env] # install unittest(pytest) related packages pip install DI-engine[test] # enable numba acceleration pip install DI-engine[fast] # install multi extra packages pip install DI-engine[common_env,test,fast] .. tip:: 某些类型的终端可能需要将安装包名称的整体,使用引号注释后才能生效该指令,如下所示: .. code-block:: bash pip install 'DI-engine[common_env,test,fast]' .. note:: 取决于安装内容与网络状态,安装耗时一般为 30 秒左右。 如果在该过程中有一些特定的依赖项未能成功安装,您可以参考 "setup.py" 文件中的版本要求,然后手动安装它。 如果遇到难以解决的安装问题,可以在 DI-engine 的 `ISSUE 区 `_ 提问,我们会尽快解答 使用 Docker 运行 ------------------ DI-engine 的镜像可以在 `DockerHub `_ 获得。拉取镜像的方法如下: .. code-block:: bash # Download Stable release DI-engine Docker image docker pull opendilab/ding:nightly # Run Docker image docker run -it opendilab/ding:nightly /bin/bash 安装检查 -------------- 安装完毕之后,您可以使用如下 python 命令检查 DI-engine 是否可用,并查看该 DI-engine 的版本信息: .. code-block:: python import ding print(ding.__version__) 您也可以直接在终端使用 DI-engine 的命令行工具: .. code-block:: bash ding -v