本文主要是介绍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
这篇关于pytorch_quantization安装的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!