安装

选择您的偏好并运行安装命令。

操作系统: Linux macOS Windows

版本: 稳定版 每夜构建版 源代码

稳定版发布。
每夜构建版,包含最新功能。
从源代码安装 GluonCV。

后端: 原生 CUDA MKL-DNN CUDA + MKL-DNN

CPU 的内置后端。
在 Nvidia GPU 上运行所需。
加速 Intel CPU 性能。
同时启用 Nvidia GPU 和 Intel CPU 加速。

先决条件

  • 需要 pip >= 9.

  • 请注意,您可以通过将“pip install gluoncv”替换为“pip install gluoncv[full]”,一次性安装所有额外的可选依赖项。

  • 每夜构建版为爱好者提供最新功能。

命令

# for mxnet
pip install --upgrade mxnet
# for pytorch
pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

pip install --upgrade gluoncv
# Here we assume CUDA 10.2 is installed. You can change the number
# according to your own CUDA version.

# for mxnet
pip install --upgrade mxnet-cu102
# for pytorch
pip install torch==1.6.0 torchvision==0.7.0

pip install --upgrade gluoncv
pip install --upgrade mxnet-mkl gluoncv
# Here we assume CUDA 10.2 is installed. You can change the number
# according to your own CUDA version.
pip install --upgrade mxnet-cu102mkl gluoncv
# for mxnet
pip install --upgrade mxnet -f https://dist.mxnet.io/python/all
# for pytorch
pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html

pip install --pre --upgrade gluoncv
# mxnet
pip install --upgrade mxnet-cu102 -f https://dist.mxnet.io/python/all
# pytorch
pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html

pip install --pre --upgrade gluoncv
pip install --pre --upgrade gluoncv
pip install --upgrade mxnet-mkl -f https://dist.mxnet.io/python/all
pip install --pre --upgrade gluoncv
pip install --upgrade mxnet-cu102mkl -f https://dist.mxnet.io/python/all
# mxnet
pip install --upgrade mxnet -f https://dist.mxnet.io/python/all
# pytorch
pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html

git clone https://github.com/dmlc/gluon-cv
cd gluon-cv && python setup.py install --user
# mxnet
pip install --upgrade mxnet-cu100 -f https://dist.mxnet.io/python/all
# pytorch
pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html

git clone https://github.com/dmlc/gluon-cv
cd gluon-cv && python setup.py install --user
pip install --upgrade mxnet-mkl -f https://dist.mxnet.io/python/all
git clone https://github.com/dmlc/gluon-cv
cd gluon-cv && python setup.py install --user
pip install --upgrade mxnet-cu102mkl -f https://dist.mxnet.io/python/all
git clone https://github.com/dmlc/gluon-cv
cd gluon-cv && python setup.py install --user

请查阅 安装 获取更多安装说明和选项。