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