Update README.md

README.md

@@ -3,7 +3,7 @@
 torch-sparse是PyTorch的一个扩展库，用于处理稀疏数据。它提供了一些功能强大的稀疏矩阵操作，以及能够高效执行稀疏计算的函数和工具。
 
 ## 依赖安装
-+ pytorch1.10或者pytorch1.13 （建议dtk-22.04.2、dtk-23.04与dtk-23.10）
++ pytorch1.10或者pytorch1.13 以及对应的torchvision（建议dtk-22.04.2、dtk-23.04与dtk-23.10）
 + python 3.7-3.10
 
 ### 1、使用源码编译方式安装
@@ -13,9 +13,10 @@ torch-sparse是PyTorch的一个扩展库，用于处理稀疏数据。它提供
 
 1. 基于光源pytorch基础镜像环境：镜像下载地址：[https://sourcefind.cn/#/image/dcu/pytorch](https://sourcefind.cn/#/image/dcu/pytorch)，根据pytorch、python、dtk及系统下载对应的镜像版本。
 
-2. 基于现有python环境：安装pytorch，pytorch whl包下载目录：[https://cancon.hpccube.com:65024/4/main/pytorch](https://cancon.hpccube.com:65024/4/main/pytorch/dtk23.04)，根据python、dtk版本,下载对应pytorch的whl包。安装命令如下：
+2. 基于现有python环境：安装pytorch和torchvision，whl包下载目录：[https://cancon.hpccube.com:65024/4/main/pytorch](https://cancon.hpccube.com:65024/4/main/pytorch)，[https://cancon.hpccube.com:65024/4/main/vision](https://cancon.hpccube.com:65024/4/main/vision)，根据python、dtk版本,下载对应pytorch和torchvision的whl包。安装命令如下：
 ```shell
 pip install torch* (下载的torch的whl包)
+pip install torchvision* (下载的torchvision的whl包)
 pip install setuptools==59.5.0 wheel
 ```