解决集群服务器异构资源下的tensorflow 多版本冲突问题

最新推荐文章于 2025-05-31 22:30:00 发布

holomain

最新推荐文章于 2025-05-31 22:30:00 发布

阅读量327

点赞数

CC 4.0 BY-SA版权

分类专栏： TensorFlow

kill yak-shaving

本文链接：https://blog.youkuaiyun.com/qq_27882063/article/details/89637532

TensorFlow 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种用于检测服务器是否配备GPU，并据此选择进入tensorflow-gpu或tensorflow-cpu虚拟环境的bash脚本示例。该脚本适用于tensorflow object detection api的模型导出过程，确保在合适的计算资源下运行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

解决：根据服务器上上的计算资源类型进入不同的的虚拟环境后再执行相应的脚本

以 tensorflow object detection api 的模型导出部分脚本化为例：

高亮部分就是判断当前主机是否有GPU，有就进入tensorflow-gpu版本的虚拟环境，
不然就是进入cpu版本的环境

#!/bin/bash
show_usage=“args: [–pipeline_config_path, --trained_checkpoint_prefix, --output_directory]”
source /etc/profile

lspci | grep -i nvidia &> /dev/null
if [ $? -eq 0 ];then
echo "有显卡"
source activate cvtf
else
echo "没显卡"
source activate cpu_cvtf
fi

INPUT_TYPE=image_tensor
PIPELINE_CONFIG_PATH= ${1} TRAINED_CKPT_PREFIX=$ {2}
EXPORT_DIR= ${3} python_path=`/usr/bin/which python` dname=$ (dirname “$PWD”)

$python_path $KaTeX parse error: Expected 'EOF', got '\ ' at position 89: …=image_tensor \̲ ̲--pipeline_conf…$ {PIPELINE_CONFIG_PATH}
–trained_checkpoint_prefix= $KaTeX parse error: Expected 'EOF', got '\ ' at position 24: …_CKPT_PREFIX} \̲ ̲--output_direct…$ {EXPORT_DIR}