latte

README.md

@@ -28,8 +28,9 @@
 
     pip install torch-2.1.0a0%2Bgit793d2b5.abi0.dtk2310-cp310-cp310-manylinux2014_x86_64.whl  (whl.zip文件中)
 
+    pip install torchvision-0.16.0+git267eff6.abi0.dtk2310.torch2.1.0-cp310-cp310-linux_x86_64.whl
+
     pip install -r requirements.txt
-    pip install torchvision==0.16.1 --no-deps
     pip install timm --no-deps
 
 ### Dockerfile（方法二）
@@ -41,8 +42,9 @@
 
     pip install torch-2.1.0a0%2Bgit793d2b5.abi0.dtk2310-cp310-cp310-manylinux2014_x86_64.whl  (whl.zip文件中)
 
+    pip install torchvision-0.16.0+git267eff6.abi0.dtk2310.torch2.1.0-cp310-cp310-linux_x86_64.whl
+
     pip install -r requirements.txt
-    pip install torchvision==0.16.1 --no-deps
     pip install timm --no-deps
 
 
@@ -50,9 +52,8 @@
 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装：
 https://developer.hpccube.com/tool/
 
-    DTK驱动：dtk23.10
-    python：python3.10
-    torch:2.1.0 (whl.zip文件中)
+    DTK驱动:     dtk23.10
+    python:      python3.10
 
 Tips：以上dtk驱动、python、torch等DCU相关工具版本需要严格一一对应
 
@@ -60,10 +61,12 @@ Tips：以上dtk驱动、python、torch等DCU相关工具版本需要严格一
 
     pip install -r requirements.txt
 
-    pip install torchvision==0.16.1 --no-deps
-
     pip install timm --no-deps
 
+    pip install torch-2.1.0a0%2Bgit793d2b5.abi0.dtk2310-cp310-cp310-manylinux2014_x86_64.whl  (whl.zip文件中)
+
+    pip install torchvision-0.16.0+git267eff6.abi0.dtk2310.torch2.1.0-cp310-cp310-linux_x86_64.whl
+
 
 ## 数据集