安装RecBole

RecBole可以通过pip, Conda和源文件三种途径安装。


系统要求

RecBole可以在以下几种系统上运行:

  • Linux
  • Windows 10
  • macOS X

RecBole需要在python 3.7或更高的环境下运行。

RecBole要求torch版本在1.7.0及以上,如果你想在GPU上运行RecBole,请确保你的CUDA版本或CUDAToolkit版本在9.2及以上。这需要你的NVIDIA驱动版本为396.26或以上(在linux系统上)或者为397.44或以上(在Windows10系统上)。


从Conda安装

如果你还没有安装Conda,可以安装Miniconda或完整的Anaconda。 如果你在中国大陆,我们推荐你使用清华镜像安装Conda。

当你完成Conda的安装后,你可以将RecBole安装在python 3.7的Conda环境中,在命令行中运行 conda create -n recbole python=3.7来创建一个新的环境。通过执行conda activate recbole来激活环境。 当Conda环境被激活后,运行如下指令来安装RecBole:

conda install -c aibox recbole

从pip安装

你只需要输入如下命令,即可从pip上安装RecBole:

pip install recbole

从源文件安装

从GitHub上下载源文件。

git clone https://github.com/RUCAIBox/RecBole.git && cd RecBole

运行如下指令进行安装:

pip install -e . --verbose

你可以阅读快速上手来获取更多关于如何快速使用的信息。