软件版本:
ubuntu16.0.4 + GTX1060 + cuda9.0 + cudnn7.4 + tensorRT5
1. 安装显卡驱动
参考博主前一篇博客 https://blog.csdn.net/weixin_37669089/article/details/85240943
2. 安装cuda9.0
对于要安装tensorRT的朋友,请使用deb文件来安装cuda,使用.run文件安装时,安装tensorRT会出错。
官网下载文件cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt-get update