Skip to content

GitLab

Commit 8f9f3e1b authored by chenxj's avatar chenxj
Browse files

update README

parent e2a56c12
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 6 deletions
README.md
View file @ 8f9f3e1b
...@@ -25,14 +25,10 @@ torchvision.datasets.CIFAR10 ...@@ -25,14 +25,10 @@ torchvision.datasets.CIFAR10
## 环境配置 ## 环境配置
在[光源](https://sourcefind.cn/#/main-page)可拉取剪枝及微调的docker镜像，在[光合开发者社区](https://cancon.hpccube.com:65024/4/main/)可下载torch安装包。mobilenet_prune推荐的镜像如下： 在[光源](https://sourcefind.cn/#/main-page)可拉取剪枝及微调的docker镜像，在[光合开发者社区](https://cancon.hpccube.com:65024/4/main/)可下载torch安装包。mobilenet_prune推荐的镜像如下：
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py37-latest docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04.1-py37-latest
``` ```
获取最新的dtk并安装依赖 安装依赖
``` ```
cd /opt
wget http://10.0.50.210:8000/jenkins/rocm/23.04.1/centos7/DTK-23.04.1-rc4-centos7-x86_64.tar.gz
tar -zxvf DTK-23.04.1-rc4-centos7-x86_64.tar.gz
source /opt/dtk-23.04.1/env.sh
cd mobilenet_prune cd mobilenet_prune
pip3 install -r requirements.txt pip3 install -r requirements.txt
``` ```
......
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