如何包装tensorflowgpu模型在大多数机器上运行？

2024-05-14 04:02:22 发布

您现在位置：Python中文网/ 问答频道 /正文
538 3

Tensorflow GPU版本、nvidia驱动程序版本、cudnn版本具有特殊的兼容性矩阵。这个兼容性矩阵在打包和分发tensorflow模型时产生了问题，当我传递tensorflow模型时，它可以由其他人运行而不受任何麻烦。基于容器的技术（例如docker）也会有问题，因为它不知道nvidia驱动程序的版本。我想知道是否有人知道最好的方式来包装tensorflow模型，它可以根据linux系统上的底层nvidia驱动程序自动配置。我怎样才能做到这一点


Tags： docker模型版本gpulinuxtensorflow系统方式
1条回答
网友
1楼 · 发布于 2024-05-14 04:02:22

最简单的方法是将tensorflow图像与docker一起使用 https://hub.docker.com/r/tensorflow/tensorflow/ 对于GPU，你可以使用英伟达工具，预装现有的图像或添加到码头工人文件…

相关问题 更多 >

    编程相关推荐