官方安装步骤:
pip install nvidia-pyindex
pip install pytorch-quantization
直接安装pytorch-quantization会找不到,需要首先安装 nvidia-pyindex 包, nvidia-pyindex是一个 pip 源,用来连接英伟达的服务器下载需要的包。
如果pip install nvidia-pyindex安装失败,可以手动将nvidia-pyindex源添加到pip源中:
pip config set global.index-url https://pypi.ngc.nvidia.com
会将pip\pip.ini文件的内容修改成:
[global]
index-url = https://pypi.ngc.nvidia.com
这时pip install pytorch_quantization会依次提示找不到 依赖包:absl-py>=0.7.0、sphinx-glpi-theme、 prettytable
可以将pip源更换后安装依赖包后再将pip源切换到nvidia-pyindex进行pytorch-quantization的安装。
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple