README增加OOM部分

README.md

@@ -23,13 +23,14 @@ TimesFM在真实世界的大型时间序列语料库上进行了预训练，可
 </div>
 
 ## 环境配置
--v 路径、docker_name和imageID根据实际情况修改
+注意：本仓库中的tensorflow只应用于长期基准测试读取数据集，运行源码发生了OOM错误，目前认为是tf-gpu与jax-gpu的内存分配冲突导致:-[GPU memory allocation — JAX documentation](https://jax.readthedocs.io/en/latest/gpu_memory_allocation.html)
+解决方法：
 ```
-注意：本仓库中的tensorflow只应用于长期基准测试读取数据集，运行源码发生了OOM错误，目前认为是tf-gpu与jax-gpu的内存分配冲突导致，解决方法：
 1、安装tf-gpu，但在timesfm/experiments/long_horizon_benchmarks/data_loader.py中添加：
 tf.config.experimental.set_visible_devices([], "GPU")
 2、安装tf-cpu
 ```
+-v 路径、docker_name和imageID根据实际情况修改
 ### Docker（方法一）
 ```
 docker pull image.sourcefind.cn:5000/dcu/admin/base/jax:0.4.23-ubuntu20.04-dtk24.04-py310