为了模型小型化，效率更高，使用TensorRT进行优化。前提是你必须要安装pycuda，可是费了我一番功夫。做一个笔记如下：
1.参考网址：
https://wiki.tiker.net/PyCuda/Installation/Linux/
2.配置代码
# 注意是python 是2.7/3.5
./configure.py --python-exe=/usr/bin/python3 --cuda-root=/usr/local/cuda-9.0 --cudadrv-lib-dir=/usr/lib/x86_64-linux-gnu --boost-inc-dir=/usr/include --boost-lib-dir=/usr/lib --boost-python-libname=boost_python-py35 --boost-thread-libname=boost_thread --no-use-shipped-boost python3 configure.py --cuda-root=/usr/local/cuda-9.0 sudo python3 setup.py install make -j 8 sudo pip3 install . 第二版 $ cd pycuda-VERSION
$ ./configure.py --python-exe=/usr/bin/python3 --cuda-root=/usr/local/cuda --cudadrv-lib-dir=/usr/lib --boost-inc-dir=/usr/include --boost-lib-dir=/usr/lib --boost-python-libname=boost_python-mt-py32 --boost-thread-libname=boost_thread-mt --no-use-shipped-boost
$ make -j 4
$ sudo python3 setup.py install