docker容器测试环境构建(docker容器内安装TensorRT的问题)
docker容器测试环境构建
docker容器内安装TensorRT的问题Ubuntu上卸载已安装的版本:
sudo apt-get purge "libnvinfer*"
如果想把安装文件都删掉以释放空间,执行:
dpkg -l | grep tensorrt
查看到安装包名,假如是nv-tensorrt-repo-ubuntu1804-cuda11.1-trt7.2.1.6-ga-20201007,则执行:
sudo dpkg -P nv-tensorrt-repo-ubuntu1804-cuda11.1-trt7.2.1.6-ga-20201007
这样就把/var/nv-tensorrt-repo-ubuntu1804-cuda11.1-trt7.2.1.6-ga-20201007_1-1这样的目录文件删掉了。
在Ubuntu主机上安装TensorRT按照https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing-debian 做即可,下载安装repo的deb文件后执行下面的步骤即可,假如安装适配CUDA10.2的TensorRT7.2.1版,执行:
sudo dpkg -i nv-tensorrt-repo-ubuntu1804-cuda10.2-trt7.2.1.6-ga-20201006_1-1_amd64.deb sudo apt-key add /var/nv-tensorrt-repo-cuda10.2-trt7.2.1.6-ga-20201006/7fa2af80.pub sudo apt-get update sudo apt-get install tensorrt
如果是在docker容器内安装就不能使用上面的步骤了,否则在安装最后一步时总是会遇到类似如下的错误:
The following packages have unmet dependencies:
tensorrt : Depends: libnvinfer-dev (= 7.2.1-6+cuda10.2) but 8.0.0-1+cuda11.3 is to be installed
Depends: libnvinfer-plugin-dev (= 7.2.1-6+cuda10.2) but 8.0.0-1+cuda11.3 is to be installed
Depends: libnvparsers-dev (= 7.2.1-6+cuda10.2) but 8.0.0-1+cuda11.3 is to be installed
Depends: libnvonnxparsers-dev (= 7.2.1-6+cuda10.2) but 8.0.0-1+cuda11.3 is to be installed
Depends: libnvinfer-samples (= 7.2.1-6+cuda10.2) but it is not going to be installed
E: Unable to correct problems, you have held broken packages.
容器内分明安装的是CUDA10.2,TensorRT也是7.2.1.6+cuda10.2版的,但是就总是报上面类似的错误,换成其他版本也是一样的继续报类似错,折腾得很郁闷,后来仔细看NVIDIA的文档发现
看来在docker容器内安装TensorRT得用另外的办法,但是呢,用Network Repo方式安装去指定安装一堆libnvinfer*库(根据需要修改8为7)是在是麻烦:
而且装完后后来编译程序时发现还有这样那样的问题,哪怕是指定版本的方式也是同样有问题:
又来回折腾了一阵还是有错,最后往下看发现了这个最简单可靠的方式:
也就是说,最前面报那种莫名其妙的错,其实是因为docker容器内配置有NVIDIA CUDA network repository,而手工下载安装nv-tensorrt-repo-ubuntu1804-cuda10.2-trt7.2.1.6-ga-20201006_1-1_amd64.deb这样的TensorRT local repository后,使用sudo apt-get install tensorrt安装的话由于有两种源会导致包冲突,只需要在/etc/apt/preferences.d/下增加个文件local-repo,内容是:
Package: * Pin: origin "" Pin-Priority: 1001
然后执行
sudo apt-get update sudo apt-get install tensorrt
就顺利安装完成local repo对应的版本cuda10.2-trt7.2.1.6,然后编译程序也没见什么错误了!
感觉NVIDIA的文档有的地方写得真的是烂,不能针对每种环境安装分别清楚无误完整地说明白么,非要把很多搅合在一起让用户自己去蹚一个个坑
另外,如果是使用的RTX30序列的GPU,CUDA版本需要 >= 11.1,不然,虽然TensorRT可以安装,但是实际运行时可能会报下面这样的错误:
[W] [TRT] Half2 support requested on hardware without native FP16 support, performance will be negatively affected.
[E] [TRT] ../rtSafe/cuda/caskUtils.cpp (98) - Assertion Error in trtSmToCask: 0 (Unsupported SM.)
11.1.0虽然可以用,但是编译caffe或darknet等某些程序时可能会有下面的错误:
nvcc fatal: Unsupported gpu architecture 'compute_86'
所以对于RTX30序列的GPU,最好安装CUDA的版本 >= 11.1.1
以上就是docker容器内安装TensorRT的问题的详细内容,更多关于docker安装TensorRT的资料请关注开心学习网其它相关文章!
- dubbo和项目搭建(使用docker部署dubbo项目的方法步骤)
- docker如何查看容器目录(docker 查看容器的挂载目录操作)
- dockercompose设置系统环境变量(使用docker compose搭建consul集群环境的例子)
- 怎么用docker部署springboot项目(Spring Boot使用Docker分层打包的设置方法)
- docker自动入门教程(Docker Machine深入详解)
- docker部署go项目(Gogs+Jenkins+Docker 自动化部署.NetCore的方法步骤)
- elasticsearch docker(在Docker中安装Elasticsearch7.6.2的教程)
- dockerfile深度解析(Dockerfile文本文件使用方法实例解析)
- docker安装nginx如何配置(docker部署nginx并且挂载文件夹和文件操作)
- docker 容器运行的数据在哪里(docker容器的几种存储详解)
- docker 退出容器详解(Docker 使用nsenter工具进入容器的操作)
- 用docker搭建在线开发环境(Docker环境搭建的简单方法)
- docker-compose项目部署(使用Docker Compose部快速署ELK亲测有效)
- docker可以快速的创建和删除(Docker 命令自动补全的实现)
- volume怎么添加docker(docker volume删除卷的操作)
- docker容器启动失败怎么查找原因(解决docker安装后运行hello-world报错的问题)
- 宋轶除了演过于曼丽,原来还演过一个青楼女子(宋轶除了演过于曼丽)
- 赵丽颖第一部当女主的戏,主角配角个个都是实力演员(赵丽颖第一部当女主的戏)
- 乾隆为何这么喜爱白塔原因是什么(乾隆为何这么喜爱白塔原因是什么)
- 逐渐消失的东北八大怪现象,进步的社会里我们遗失的是什么(逐渐消失的东北八大怪现象)
- 今日大雪,大雪养生这样做,一年都有好气色(大雪养生这样做)
- 粗盐是什么 粗盐的功效与作用(粗盐的功效与作用)
热门推荐
- vuejs组件使用教程交流(Vue vee-validate插件的简单使用)
- dockernginx怎么设置容器(docker nginx + https 子域名配置详细教程)
- dede织梦的后台如何调整和使用(dedecms织梦模板去掉文档内链自动连接关键词的下划线)
- 使用Fiddler测试WebApi接口
- python 多线程与多进程(python 多线程串行和并行的实例)
- 如何使java与sqlserver数据库连接(java连接mysql数据库 java连接sql server数据库)
- 如何清除mysql的binlog
- HTMl5 sessionStorage和localStorage
- laravel后台框架(laravel-admin的多级联动方法)
- python ssh登录服务器(python利用跳板机ssh远程连接redis的方法)
排行榜
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9