update

README.md

@@ -26,16 +26,11 @@
 
     docker run --shm-size 10g --network=host --name=graphcast --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v 项目地址(绝对路径):/home/ -v /opt/hyhal:/opt/hyhal:ro -it <your IMAGE ID> bash
 
+    pip install -r requirements.txt --constraint constraints.txt
     pip install -e .
     pip uninstall shapely
-
-    pip install jupyter 
     pip install shapely
 
-    pip install google-cloud
-    pip install google-cloud-vision
-    pip install protobuf
-
     pip install --upgrade google-api-python-client
     pip install google.cloud.bigquery
     pip install google.cloud.storage
@@ -47,16 +42,11 @@
 
     docker run --shm-size 10g --network=host --name=graphcast --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v 项目地址(绝对路径):/home/ -v /opt/hyhal:/opt/hyhal:ro -it <your IMAGE ID> bash
 
+    pip install -r requirements.txt --constraint constraints.txt
     pip install -e .
     pip uninstall shapely
-
-    pip install jupyter 
     pip install shapely
 
-    pip install google-cloud
-    pip install google-cloud-vision
-    pip install protobuf
-
     pip install --upgrade google-api-python-client
     pip install google.cloud.bigquery
     pip install google.cloud.storage
@@ -71,27 +61,34 @@ Tips：以上dtk驱动、python、jax等DCU相关工具版本需要严格一一
 
 2、其他非特殊库
 
+    pip install -r requirements.txt --constraint constraints.txt
     pip install -e .
     pip uninstall shapely
-    pip install jupyter 
     pip install shapely
 
+    pip install --upgrade google-api-python-client
+    pip install google.cloud.bigquery
+    pip install google.cloud.storage
+
     
 ## 数据集
 
-https://console.cloud.google.com/storage/browser/dm_graphcast
+[Google Cloud](https://console.cloud.google.com/storage/browser/dm_graphcast) | [SCNet](http://113.200.138.88:18080/aidatasets/project-dependency/dm_graphcast)
 
-注意：该数据集按需下载，在执行`graphcast_demo.ipynb`时，可自动下载示例数据。
 
+注意：该数据集按需下载，在执行`graphcast_demo.ipynb`时，可自动下载示例数据。
 
 ## 训练
 
 参考并执行`graphcast_demo.ipynb`中的`Train Model`部分。
 
+注意：使用`jupyter notebook --no-browser --ip=0.0.0.0 --allow-root`启动服务以运行代码。
+
 ## 推理
 
 参考并执行`graphcast_demo.ipynb`中的`Run the model`部分。
 
+
 ## result
 
 ![alt text](asset/result.png)
@@ -110,9 +107,13 @@ https://console.cloud.google.com/storage/browser/dm_graphcast
 
 `气象,交通,环境`
 
+## 预训练权重
+
+[Google Cloud](https://console.cloud.google.com/storage/browser/dm_graphcast) | [SCNet](http://113.200.138.88:18080/aimodels/findsource-dependency/dm_graphcast) 高速下载通道 
+
 ## 源码仓库及问题反馈
 
-* https://developer.hpccube.com/codes/modelzoo/cfd_jax
+* https://developer.hpccube.com/codes/modelzoo/graphcast_jax
 
 ## 参考资料
 

constraints.txt

+jax==0.4.23
+jaxlib==0.4.23
\ No newline at end of file

requirements.txt

+cartopy
+chex
+colabtools
+dask
+dm-haiku
+dm-tree
+jraph
+matplotlib
+pandas
+rtree
+scipy
+trimesh
+typing_extensions
+xarray
+numpy==1.26.4
+jupyter
+google-cloud
+google-cloud-vision
+protobuf
\ No newline at end of file

setup.py

@@ -30,22 +30,22 @@ setup(
     url="https://github.com/deepmind/graphcast",
     packages=["graphcast"],
     install_requires=[
-        "cartopy",
-        "chex",
-        "colabtools",
-        "dask",
-        "dm-haiku",
-        "dm-tree",
-        "jax",
-        "jraph",
-        "matplotlib",
-        "numpy",
-        "pandas",
-        "rtree",
-        "scipy",
-        "trimesh",
-        "typing_extensions",
-        "xarray",
+        # "cartopy",
+        # "chex",
+        # "colabtools",
+        # "dask",
+        # "dm-haiku",
+        # "dm-tree",
+        # # "jax",
+        # "jraph",
+        # "matplotlib",
+        # # "numpy",
+        # "pandas",
+        # "rtree",
+        # "scipy",
+        # "trimesh",
+        # "typing_extensions",
+        # "xarray",
     ],
     classifiers=[
         "Development Status :: 3 - Alpha",