我正在尝试建立一个自定义的码头形象,以服务我们的图像分类模型。
在Google云上使用Ubuntu18.04。GPU型Nvidia-t4。在同一台机器上,使用Tensorflow - GPU 1.9.0并按预期工作。当我使用命令构建docker文件时:
sudo nvidia-docker build -t name .
看到以下错误消息。模型加载在CPU上而不是GPU上,推理在CPU上运行。
2021-01-05 20:46:59.617414: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports in
任何有使用vast.ai进行云计算的经验的人都知道,当租用多个GPU时,您需要做一些设置来利用额外的GPU吗?
因为当租用6或8个GPU而不是仅仅一个GPU时,我不会注意到速度上的任何差异。我刚开始使用vast.ai进行云计算。
我使用的是默认的码头:用于深度学习框架TensorFlow ()的正式对接图像。
成功加载了tensorflow/tensorflow:夜-GPU-py3
然后再安装keras:
pip install keras
我还使用此方法检查了可用的GPU,所有GPU都被正确检测到:
from keras import backend as K
K.tensorflow_b