Skip to content

GitLab

Commit 65b57d01 authored by dcuai's avatar dcuai
Browse files

Update dtk24.04.1镜像

parent 2691968a
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
README.md
View file @ 65b57d01
...@@ -23,8 +23,8 @@ mv neuralforecast-itransformer_pytorch neuralforecast # 去框架名后缀 ...@@ -23,8 +23,8 @@ mv neuralforecast-itransformer_pytorch neuralforecast # 去框架名后缀
### Docker（方法一） ### Docker（方法一）
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk23.10-py38 docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
# <your IMAGE ID>为以上拉取的docker的镜像ID替换，本镜像为：ffa1f63239fc # <your IMAGE ID>为以上拉取的docker的镜像ID替换
docker run -it --shm-size=32G -v $PWD/neuralforecast:/home/neuralforecast -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name neuralforecast <your IMAGE ID> bash docker run -it --shm-size=32G -v $PWD/neuralforecast:/home/neuralforecast -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name neuralforecast <your IMAGE ID> bash
cd /home/neuralforecast cd /home/neuralforecast
pip install -r requirements.txt # requirements.txt pip install -r requirements.txt # requirements.txt
...@@ -40,8 +40,8 @@ docker run --shm-size=32G --name neuralforecast -v /opt/hyhal:/opt/hyhal:ro --pr ...@@ -40,8 +40,8 @@ docker run --shm-size=32G --name neuralforecast -v /opt/hyhal:/opt/hyhal:ro --pr
1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装： 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装：
- https://developer.hpccube.com/tool/ - https://developer.hpccube.com/tool/
``` ```
DTK驱动:dtk23.10 DTK驱动:dtk24.04.1
python:python3.8 python:python3.10
torch:2.1.0 torch:2.1.0
torchvision:0.16.0 torchvision:0.16.0
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment