為了模型小型化，效率更高，使用TensorRT進行優化。前提是你必須要安裝pycuda，可是費了我一番功夫。做一個筆記如下：
1.參考網址：
https://wiki.tiker.net/PyCuda/Installation/Linux/
2.配置代碼
1 # 注意是python 是2.7/3.5 2 ./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 3 4 python3 configure.py --cuda-root=/usr/local/cuda-9.0 5 6 sudo python3 setup.py install 7 8 make -j 8 9 10 sudo pip3 install . 11 12 13 14 第二版 15 16 $ cd pycuda-VERSION 17 $ ./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 18 $ make -j 4 19 $ sudo python3 setup.py install