upgrade torch and triton (#3677)

* upgrade deps

* upgrade torchvision requirements

Author: grimoire

lmdeploy/pytorch/check_env/triton.py

@@ -3,7 +3,7 @@
 
 from .base import BaseChecker
 
-MAX_TRITON_VERSION = '3.2.0'
+MAX_TRITON_VERSION = '3.3.1'
 MIN_TRITON_VERSION = '3.0.0'
 
 

requirements/runtime_cuda.txt

@@ -18,8 +18,8 @@ safetensors
 sentencepiece
 shortuuid
 tiktoken
-torch<=2.6.0,>=2.0.0
-torchvision<=0.21.0,>=0.15.0
+torch<=2.7.1,>=2.0.0
+torchvision<=0.22.1,>=0.15.0
 transformers
-triton<=3.2.0,>=3.0.0; sys_platform == "linux"
+triton<=3.3.1,>=3.0.0; sys_platform == "linux"
 uvicorn