update ci

ci/test.sh

@@ -2,6 +2,7 @@ set -ex
 python3 -m pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
 python3 -m pip install --user --upgrade pip
 python3 -m pip install --user flake8 pytest
+python3 -m pip install nvidia-pyindex
 if [ -f requirements.txt ]; then pip install --user -r requirements.txt; fi
 python3 -m pip install oneflow --user -U -f https://staging.oneflow.info/branch/master/cu102
 python3 setup.py install

requirements.txt

+nvidia-tensorrt==7.2.3.4
+pycuda
 onnx>=1.8.0
 onnx-simplifier>=0.3.3
 onnxoptimizer>=0.2.5
@@ -7,4 +9,4 @@ paddlepaddle==2.0.1
 tensorflow_cpu==2.4.0
 torch==1.8.1
 torchvision==0.9.1
-paddle2onnx>=0.5
\ No newline at end of file
+paddle2onnx>=0.5