From 1acd42f9903c1ed15e0c208153008445bc16fe6c Mon Sep 17 00:00:00 2001
From: sunyi001 <1659275352@qq.com>
Date: Thu, 26 Jun 2025 09:44:14 +0800
Subject: [PATCH] Modify the installation method of torchvision

---
 PyTorch/built-in/rl/VeRL_for_PyTorch/README.md | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/PyTorch/built-in/rl/VeRL_for_PyTorch/README.md b/PyTorch/built-in/rl/VeRL_for_PyTorch/README.md
index a3d5acd1ad..e4d0d5823c 100644
--- a/PyTorch/built-in/rl/VeRL_for_PyTorch/README.md
+++ b/PyTorch/built-in/rl/VeRL_for_PyTorch/README.md
@@ -108,14 +108,9 @@ verl‌是一个集SFT（监督学习）与RL（强化学习）于一体的灵
   cd ..
   ```
 
-- 对于VL模型，需要安装torchvision，克隆torchvision仓并切换到v0.20.1
+- 对于VL模型，需要安装torchvision
   ```shell
-  git clone -b v0.20.1 --depth 1  https://github.com/pytorch/vision.git
-  cd vision
-  python setup.py bdist_wheel
-  # 安装`torchvision`前，需要先执行`pip uninstall torchvision`卸载原来的`torchvision`，如果环境中有`triton`，需要执行`pip uninstall triton`卸载
-  pip install dist/*.whl
-  cd ..
+  pip install torchvision==0.20.1+cpu --index-url https://download.pytorch.org/whl/cpu
   ```
 
 - 环境准备指导。
@@ -335,3 +330,9 @@ verl‌是一个集SFT（监督学习）与RL（强化学习）于一体的灵
   export GLOO_SOCKET_IFNAME=网口名称
   export HCCL_SOCKET_IFNAME=网口名称 
   ```
+
+- 如果遇到`ImportError: cannot import name "fused_topk" from "fused_moe_module" (unknown location)`错误，请使用以下命令卸载`triton`：
+
+  ```shell
+  pip uninstall triton
+  ```
-- 
Gitee