TorchRec 安装和配置指南

TorchRec 安装和配置指南

torchrec Pytorch domain library for recommendation systems 项目地址: https://gitcode.com/gh_mirrors/to/torchrec

1、项目的基础介绍和主要的编程语言

TorchRec 是一个由 PyTorch 开发的领域库，专门用于推荐系统。它提供了推荐系统所需的各种稀疏性和并行性原语，使得大规模推荐系统的训练和推理变得更加高效。TorchRec 主要使用 Python 编程语言，并且依赖于 PyTorch 框架。

2、项目使用的关键技术和框架

TorchRec 使用的关键技术和框架包括：

• PyTorch: 作为底层框架，提供深度学习模型的构建和训练支持。
• FBGEMM: 用于推荐系统的优化内核，提供高效的矩阵运算。
• CUDA: 如果使用 GPU 进行加速，需要安装 CUDA 支持。
• Python: 主要编程语言，用于编写和运行推荐系统模型。

3、项目安装和配置的准备工作和详细的安装步骤

准备工作

在开始安装 TorchRec 之前，请确保你的系统满足以下要求：

• Python 3.7 或更高版本：TorchRec 需要 Python 3.7 或更高版本。
• PyTorch: 需要安装最新版本的 PyTorch。
• CUDA（可选）：如果你计划使用 GPU 进行加速，需要安装 CUDA 11.8 或更高版本。
• Git: 用于克隆 TorchRec 的代码库。

安装步骤

1. 安装 PyTorch

首先，你需要安装 PyTorch。根据你的系统配置选择合适的安装命令：

• CUDA 12.1:

  pip install torch --index-url https://download.pytorch.org/whl/nightly/cu121
  

• CUDA 11.8:

  pip install torch --index-url https://download.pytorch.org/whl/nightly/cu118
  

• CPU 版本:

  pip install torch --index-url https://download.pytorch.org/whl/nightly/cpu
  

2. 克隆 TorchRec 代码库

使用 Git 克隆 TorchRec 的代码库到本地：

git clone --recursive https://github.com/pytorch/torchrec.git
cd torchrec

3. 安装 FBGEMM

根据你的系统配置选择合适的 FBGEMM 安装命令：

• CUDA 12.1:

  pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cu121
  

• CUDA 11.8:

  pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cu118
  

• CPU 版本:

  pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cpu
  

4. 安装其他依赖

安装 TorchRec 所需的其他依赖：

pip install -r requirements.txt

5. 安装 TorchRec

使用以下命令安装 TorchRec：

python setup.py install develop

6. 测试安装

为了确保安装成功，可以运行测试脚本：

• GPU 模式:

  torchx run -s local_cwd dist.ddp -j 1x2 --gpu 2 --script test_installation.py
  

• CPU 模式:

  torchx run -s local_cwd dist.ddp -j 1x2 --script test_installation.py -- --cpu_only
  

总结

通过以上步骤，你应该已经成功安装并配置了 TorchRec。现在你可以开始使用 TorchRec 构建和训练推荐系统模型了。如果在安装过程中遇到任何问题，请参考官方文档或社区支持。

torchrec Pytorch domain library for recommendation systems 项目地址: https://gitcode.com/gh_mirrors/to/torchrec