安装gpu版本的paddle和paddleclas

monster&&summit2023-09-26 8:26

c 复制代码

python -m pip install paddlepaddle-gpu==2.3.2.post111 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html

以上支持cuda11.1版本

其他需求可查阅文档在这里

1 在虚拟环境中安装所需的Python库：

c 复制代码

pip install paddlepaddle paddlehub paddleclas

2 安装ipykernel库：

c 复制代码

pip install ipykernel

或者源码安装

1 在虚拟环境中安装所需的Python库：

c 复制代码

pip install paddlepaddle paddlehub ipykernel

2 拉取paddleclas代码

从 GitHub 下载：

c 复制代码

git clone https://github.com/PaddlePaddle/PaddleClas.git -b release/2.4

如果访问 GitHub 网速较慢，可以从 Gitee 下载，命令如下：

c 复制代码

git clone https://gitee.com/paddlepaddle/PaddleClas.git -b release/2.4

3 安装依赖：

c 复制代码

pip install -r requirements.txt -i https://mirror.baidu.com/pypi/simple

4 可能会报一个关于protobuf的错，执行下边命令，然后再执行上一条命令：

c 复制代码

pip install --upgrade protobuf===3.20.3

5 安装

c 复制代码

python setup.py install

完成