diff --git a/example/auto_compression/pytorch_yolo_series/README.md b/example/auto_compression/pytorch_yolo_series/README.md index 052daf27da29e552397b0af50300918d8e106979..9273306340164bc3b1b2728c0396f370fd9952f3 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 ```