Update README.md

0dfefcd6 · suily · 667aed44 · 0dfefcd6
Commit 0dfefcd6 authored Sep 02, 2024 by suily
Hide whitespace changes
Inline Side-by-side

Showing with 27 additions and 12 deletions

README.md README.md +27 -12

No files found.
--- a/README.md
+++ b/README.md
@@ -36,10 +36,17 @@ docker pull image.sourcefind.cn:5000/dcu/admin/base/jax:0.4.23-ubuntu20.04-dtk24
 docker run -it --network=host --privileged=true --name=docker_name --device=/dev/kfd --device=/dev/dri --group-add video --shm-size=32G  --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /path/your_code_data/:/path/your_code_data/ imageID /bin/bash

 cd /your_code_path/timesfm
+pip install praxis==1.2.0
+pip install paxml==1.2.0
 pip install -r requirements.txt
-pip install tensorflow-2.13.1+das1.0+git429d21b.abi1.dtk2404-cp310-cp310-manylinux2014_x86_64.whl（或 tensorflow-cpu==2.13.1）
-pip uninstall protobuf
-pip install protobuf==3.20.3
+wget https://cancon.hpccube.com:65024/directlink/4/tensorflow/DAS1.0/tensorflow-2.13.1+das1.0+git429d21b.abi1.dtk2404-cp310-cp310-manylinux2014_x86_64.whl #（或 tensorflow-cpu==2.13.1）
+wget https://cancon.hpccube.com:65024/directlink/4/pytorch/DAS1.0/torch-2.1.0+das1.0+git00661e0.abi0.dtk2404-cp310-cp310-manylinux2014_x86_64.whl
+wget https://cancon.hpccube.com:65024/directlink/4/jax/DAS1.1/jax-0.4.23+das1.1.git387bd43.abi1.dtk2404-py3-none-any.whl
+wget https://cancon.hpccube.com:65024/directlink/4/jax/DAS1.0/jaxlib-0.4.23+das1.0+git97306ab.abi1.dtk2404-cp310-cp310-manylinux2014_x86_64.whl
+pip install tensorflow-2.13.1+das1.0+git429d21b.abi1.dtk2404-cp310-cp310-manylinux2014_x86_64.whl #（或 tensorflow-cpu==2.13.1）
+pip install torch-2.1.0+das1.0+git00661e0.abi0.dtk2404-cp310-cp310-manylinux2014_x86_64.whl
+pip install jax-0.4.23+das1.1.git387bd43.abi1.dtk2404-py3-none-any.whl
+pip install jaxlib-0.4.23+das1.0+git97306ab.abi1.dtk2404-cp310-cp310-manylinux2014_x86_64.whl
 ```
 ### Dockerfile（方法二）
 ```
@@ -48,10 +55,17 @@ docker build --no-cache -t timesfm:latest .
 docker run -it --network=host --privileged=true --name=docker_name --device=/dev/kfd --device=/dev/dri --group-add video --shm-size=32G  --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /path/your_code_data/:/path/your_code_data/ imageID /bin/bash

 cd /your_code_path/timesfm
+pip install praxis==1.2.0
+pip install paxml==1.2.0
 pip install -r requirements.txt
-pip install tensorflow-2.13.1+das1.0+git429d21b.abi1.dtk2404-cp310-cp310-manylinux2014_x86_64.whl （或 tensorflow-cpu==2.13.1）
-pip uninstall protobuf
-pip install protobuf==3.20.3
+wget https://cancon.hpccube.com:65024/directlink/4/tensorflow/DAS1.0/tensorflow-2.13.1+das1.0+git429d21b.abi1.dtk2404-cp310-cp310-manylinux2014_x86_64.whl #（或 tensorflow-cpu==2.13.1）
+wget https://cancon.hpccube.com:65024/directlink/4/pytorch/DAS1.0/torch-2.1.0+das1.0+git00661e0.abi0.dtk2404-cp310-cp310-manylinux2014_x86_64.whl
+wget https://cancon.hpccube.com:65024/directlink/4/jax/DAS1.1/jax-0.4.23+das1.1.git387bd43.abi1.dtk2404-py3-none-any.whl
+wget https://cancon.hpccube.com:65024/directlink/4/jax/DAS1.0/jaxlib-0.4.23+das1.0+git97306ab.abi1.dtk2404-cp310-cp310-manylinux2014_x86_64.whl
+pip install tensorflow-2.13.1+das1.0+git429d21b.abi1.dtk2404-cp310-cp310-manylinux2014_x86_64.whl #（或 tensorflow-cpu==2.13.1）
+pip install torch-2.1.0+das1.0+git00661e0.abi0.dtk2404-cp310-cp310-manylinux2014_x86_64.whl
+pip install jax-0.4.23+das1.1.git387bd43.abi1.dtk2404-py3-none-any.whl
+pip install jaxlib-0.4.23+das1.0+git97306ab.abi1.dtk2404-cp310-cp310-manylinux2014_x86_64.whl
 ```
 ### Anaconda（方法三）
 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装: https://developer.hpccube.com/tool/
@@ -60,15 +74,20 @@ DTK软件栈：dtk24.04
 python：python3.10
 jax：0.4.23
 tensorflow：2.13.1 （或 tensorflow-cpu==2.13.1）
+torch：2.1.0
 ```
 `Tips：以上dtk软件栈、python、jax等DCU相关工具版本需要严格一一对应`

 2、其他非特殊库直接按照下面步骤进行安装
 ```
 cd /your_code_path/timesfm
+pip install praxis==1.2.0
+pip install paxml==1.2.0
 pip install -r requirements.txt
-pip uninstall protobuf
-pip install protobuf==3.20.3
+pip install tensorflow-2.13.1+das1.0+git429d21b.abi1.dtk2404-cp310-cp310-manylinux2014_x86_64.whl #（或 tensorflow-cpu==2.13.1）
+pip install torch-2.1.0+das1.0+git00661e0.abi0.dtk2404-cp310-cp310-manylinux2014_x86_64.whl
+pip install jax-0.4.23+das1.1.git387bd43.abi1.dtk2404-py3-none-any.whl
+pip install jaxlib-0.4.23+das1.0+git97306ab.abi1.dtk2404-cp310-cp310-manylinux2014_x86_64.whl
 ```
 ## 数据集
 基准测试数据集运行时会gluonts自动下载，长期基准测试数据集需从Google Drive手动下载（需要魔法）：
@@ -110,10 +129,6 @@ DATA_DICT = {
 -https://hf-mirror.com/google/timesfm-1.0-200m
 ```
 # "model"是存储目录，可自订
-1、通过git
-cd timesfm
-git clone https://hf-mirror.com/google/timesfm-1.0-200m model
-2、通过huggingface-cli
 cd timesfm
 export HF_DATASETS_CACHE="path/timesfm/model"
 export HF_ENDPOINT=https://hf-mirror.com  # 设置下载地址