无法安装tensorflowjs及Keras模型转tf.js格式问题求助

版本兼容问题说明 #

tensorflowjs与TensorFlow版本存在严格对应关系，直接执行pip install tensorflowjs会默认安装最新版，和TensorFlow 2.6.0产生版本冲突，导致安装报错。

具体解决步骤 #

1. 安装匹配版本的tensorflowjs
   针对TensorFlow 2.6.0，安装经过验证的兼容版本：

   pip install tensorflowjs==3.15.0
   

2. 加载模型并完成转换
   在Jupyter Notebook中执行以下代码：

   import tensorflow as tf
   import tensorflowjs as tfjs
   
   # 加载本地的Keras模型
   model = tf.keras.models.load_model("melanoma_binary_classifer.keras")
   
   # 指定转换后模型的保存目录
   tfjs_target_dir = "./tfjs_model"
   tfjs.converters.save_keras_model(model, tfjs_target_dir)
   

3. 验证转换结果
   转换完成后，目标目录会生成model.json文件和若干权重分片文件（命名格式为group1-shard*of*），这些就是可直接在TensorFlow.js中加载的模型文件。

额外环境隔离方案（可选） #

如果仍存在依赖冲突，可通过虚拟环境隔离环境：

# 创建虚拟环境
python -m venv tf26_env
# Windows激活环境
tf26_env\Scripts\activate
# Linux/macOS激活环境
source tf26_env/bin/activate
# 依次安装对应版本依赖
pip install tensorflow==2.6.0
pip install tensorflowjs==3.15.0

同时要确保Jupyter Notebook使用的是该虚拟环境的Python内核，可通过!pip list在Notebook中查看已安装包版本确认。

内容的提问来源于stack exchange，提问作者Chris