我有一个用Keras构建的情感分类器,我想用我的GPU运行。根据Tensorflows GPU support page的建议,我有installed Docker and downloaded a Tensorflow Docker image。
现在,当我尝试在其中一个Tensorflow图像上运行代码时,当我尝试导入Keras或Pandas时,我会收到错误代码。
当涉及到Docker时,我是一个新手,但据我所知,镜像根本没有安装这些库。那么,如果我想使用除了Tensorflow之外的其他东西或映像上安装的其他东西,我该怎么办?如何将这些添加到图像中?
2条答案
按热度按时间hlswsv351#
选项一:将包添加到容器:
缺点是每次重新创建容器时都必须重复此操作。
选项二:创建您自己的图像,使用TensorFlow图像作为基础
创建一个名为
Dockerfile
的文件:然后从中构建一个图像:
然后使用您自己的图像运行:
我还建议在开发环境中使用docker-compose,这样你就不必记住所有这些命令。您可以创建一个
docker-compose.yml
并使用YAML格式描述容器。然后你可以只构建docker-compose build
,运行docker-compose up
。hwazgwia2#
从我的经验来看,我总是发现创建一个通用的Docker镜像并将您的需求安装到其中要好得多。我知道你最初的问题要求使用Tensorflow Docker镜像,但我会留下这个答案供参考。下面是一个简单的Dockerfile用途:
您可以使用以下命令构建映像:
docker build -t my_image .