首页 > 其他 > 详细

TensorRT下安装pycuda

时间:2019-03-04 15:55:00      阅读:915      评论:0      收藏:0      [点我收藏+]

为了模型小型化,效率更高,使用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

 

TensorRT下安装pycuda

原文:https://www.cnblogs.com/demo-deng/p/10470734.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!