From 7050dc2b491d7d27ff6b5ced145c6d83a5956be9 Mon Sep 17 00:00:00 2001 From: leiqing <54695910+leiqing1@users.noreply.github.com> Date: Mon, 22 Aug 2022 15:24:13 +0800 Subject: [PATCH] Update README.md --- example/auto_compression/pytorch_yolo_series/README.md | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/example/auto_compression/pytorch_yolo_series/README.md b/example/auto_compression/pytorch_yolo_series/README.md index 052daf27..92733063 100644 --- a/example/auto_compression/pytorch_yolo_series/README.md +++ b/example/auto_compression/pytorch_yolo_series/README.md @@ -50,14 +50,13 @@ ``` # CPU pip install paddlepaddle==2.3.2 -# GPU -pip install paddlepaddle-gpu==2.3.2 +# GPU 以Ubuntu、CUDA 11.2为例 +python -m pip install paddlepaddle-gpu==2.3.2.post112 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html ``` -(2)安装paddleslim: +(2)安装paddleslim>=2.3.3: ```shell -git clone https://github.com/PaddlePaddle/PaddleSlim.git & cd PaddleSlim -python setup.py install +pip install paddleslim==2.3.3 ``` -- GitLab