使用REST和SSL自证书为GPU服务的Tensorflow
使用REST和SSL自证书为GPU服务的Tensorflow
提问于 2018-12-22 21:13:26
我试图在Centos 7 Docker容器中安装带有REST的TensorFlow-gpu。但我找不到确切的程序。是否需要安装下列依赖项？
- 我安装了库达9.0
- cdDNN 7.4
- NCCL 2.x
回答 2
Stack Overflow用户
发布于 2018-12-22 23:41:31
我还没有开始使用GPU构建tensorflow服务。我正处于研究阶段的中期~在这个过程中，在每一篇关于Ubuntu安装的文章中，我都试图使用install.in centos 7 .所以我没有任何码头文件。
希望这能帮你和我找到解决办法。
Stack Overflow用户
发布于 2018-12-24 06:06:20
下面是我用来构建tensorflow服务-运行时坞映像的内容。
FROM nvidia/cuda:9.0-cudnn7-runtime-centos7
ARG TF_VERSION=1.9.0
RUN yum install -y \
yum-plugin-ovl \
libgomp \
ca-certificates \
zip \
unzip \
curl \
&& \
yum clean all
WORKDIR /usr/
RUN curl -sSL -o /usr/nccl_2.2.13-1-cuda9.0_x86_64.tgz http://some-of-my-net-disk/tensorflow-serving/lib/nccl_2.2.13-1-cuda9.0_x86_64.tgz && \ # Change your way to get nccl library here
tar -xvf nccl_2.2.13-1-cuda9.0_x86_64.tgz &&\
rm -f nccl_2.2.13-1-cuda9.0_x86_64.tgz
ENV LD_LIBRARY_PATH /usr/nccl_2.2.13-1+cuda9.0_x86_64/lib/:${LD_LIBRARY_PATH}
# Change your way to get tensorflow_model_server here
WORKDIR /serving
RUN curl -sSL -o /usr/local/bin/tensorflow_model_server http://some-of-my-net-disk/tensorflow-serving/bin/tf-serving-${TF_VERSION}/tensorflow_model_server_gpu-centos &&\
chmod u+x /usr/local/bin/tensorflow_model_server对我来说，这很好。希望能帮上忙。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/53899338
复制相关文章
相似问题
腾讯云开发者
