Ubuntu18.04+CUDA9.0+cuDNN7.5.0.56+TensorRT5.1.2.2RC+pycuda2019.1.1+onnx1.4.1 安装总结

NVIDIA驱动

下载地址：​​​https://www.geforce.com/drivers#​​​

我选驱动390.116

CUDA Toolkit安装

下载地址：​​​https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal​​​

我选CUDA Toolkit 9.0，其他版本可选：​​​https://developer.nvidia.com/cuda-toolkit-archive​​​

​​sudo sh cuda_9.0.176_384.81_linux.run​​

cudnn安装

下载地址：​​​https://developer.nvidia.com/rdp/cudnn-archive​​​

我选cudnn-9.0-linux-x64-v7.5.0.56.tgz

tar zxvf cudnn-9.0-linux-x64-v7.5.0.56.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.7.5.0
sudo ln -sf libcudnn.so.7.5.0 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig

TensorRT安装

下载地址：​​​https://developer.nvidia.com/nvidia-tensorrt-download​​​

我选TensorRT-5.1.2.2.Ubuntu-16.04.4.x86_64-gnu.cuda-9.0.cudnn7.5.tar.gz

tar zxvf TensorRT-5.1.2.2.Ubuntu-16.04.4.x86_64-gnu.cuda-9.0.cudnn7.5.tar.gz
gedit ~/.bashrc

环节变量中添加解压路径+lib

source ~/.bashrc

/home/hylink/TensorRT-5.1.2.2/python/
pip install tensorrt-5.1.2.2-cp36-none-linux_x86_64.whl

pycuda安装

​​pip install pycuda==2019.1.1 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com​​

其他版本可选：​​​https://pypi.org/project/pycuda​​​

onnx安装

​​pip install onnx==1.4.1 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com​​

各个版本对应非常重要，版本错误会造成程序运行异常

功能验证

可基于​​​https://github.com/YonghaoHe/A-Light-and-Fast-Face-Detector-for-Edge-Devices/tree/master/face_detection​​​进行

其他已测试可用版本