TensorRT-开发入门
Docker For Ubuntu 16.04(心路历程)
以及在Ubuntu上Docker中使用TensorRT的心路历程,这也是我为以后像搭建出要给实实在在能用的深度学习应用而做的准备
这篇记录在Ubuntu上安装Docker,并且安装nvidia-docker的心路历程,还有NGC(Nvidia GPU Cloud )的里面的container的使用。(尤其是TensorRT container的一个使用)
一、在Ubuntu上安装docker-ce
这里的docker-ce是docker的社区版,因为社区版是不收费的,一般情况下已经够用了。先去官网,Docker For Ubuntu 这里可以找到安装的教程,我这里使用deb文件安装。
去(1)ubuntu的docker deb文件, 根据你的ubuntu的版本,查看Ubuntu版本命令,
cat /etc/issue
会显示出ubuntu的版本号,根据版本号在这个网站上(2)ubuntu版本号名字对应关系表, 然后在(1)链接中找到自己系统的对应名字,点进去名字/pool/stable/
到了这一级目录,就到了选择主机位数的时候了,不清楚的小伙伴,使用cat /proc/version
命令查看,我的是amd64。(需要注意一点就是我的ubuntu的名字是xenial)。这里有一个小小的提示,在后面安装nvidia-docker2的时候,需要docker-ce 18.06的支持,所以大家在下载的时候,需要下载这个版本的。(一张图胜过千言万语,下面是deb文件展示,上面是路径,我下载的版本是红色箭头指向的那个版本)下面就开始安装了,安装之前需要废话两句,如果你的系统上已经有了docker怎么办,那要看你是通过apt还是apt-get安装的。命令如下:
1 | 查看是否安装命令 |
图2
二、在ubuntu上安装nvidia-docker和使用NGC container
这里有一个大前提是,你的主机上已经安装过nvidia的驱动了,详细的过程请查看安装nvidia driver。
- 安装nvidia-docker(这个步骤一定要在安装docker之后)
1 | curl -s -L https://nvidia.github.io/nvidia-docker/gpkey | \ |
- 然后去注册NGC注册教程
上面的教程已经很清楚了。(最主要的是生成那个api key)
附一张注册以后的主页
这里面是所有的container
后面运行示例的时候会下载TensorRT的样例。
- 本地安装NGC image(这是Docker的镜像文件,可以当作一个application),以及运行image为container
1 | sudo docker login nvcr.io |
图3
图4
图5
三、到此我们就结束了,可能有很多小伙伴说这些太简单了,其实不简单,这些内容是我花了一整天时间,查各种文档,最后简练的将这些东西整合到一起的。如果大家有什么不明白的地方,可以直接给我发邮件或者直接加我qq,我或许可以解答你的疑惑。
qq:329804334
Email:air@weaf.top