pytorch_quantization安装

官方安装步骤:

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

你可能感兴趣的:(ADAS,python,pytorch)