Docker环境下RAPIDS pip安装失败:PyPI占位包问题求助
解决RAPIDS在Docker中pip安装的占位包问题
方案1:调整索引优先级,优先拉取NVIDIA仓库包
把NVIDIA PyPI设为主索引,默认PyPI作为额外索引,这样pip会优先从NVIDIA仓库获取RAPIDS包,依赖包则从默认仓库拉取:
pip3 install cudf-cu11 cuml-cu11 cugraph-cu11 cucim --index-url=https://pypi.nvidia.com --extra-index-url=https://pypi.org/simple
方案2:清理pip缓存后重试
如果pip缓存了默认PyPI的占位包,先清空缓存再执行安装:
pip3 cache purge pip3 install cudf-cu11 cuml-cu11 cugraph-cu11 cucim --extra-index-url=https://pypi.nvidia.com
方案3:升级pip到最新版本
旧版pip的索引优先级处理可能存在逻辑问题,升级后再尝试安装:
pip3 install --upgrade pip pip3 install cudf-cu11 cuml-cu11 cugraph-cu11 cucim --extra-index-url=https://pypi.nvidia.com
方案4:为每个RAPIDS包单独指定索引
针对每个需要安装的RAPIDS包明确绑定NVIDIA仓库,确保不会拉取到占位包:
pip3 install \ cudf-cu11 --extra-index-url=https://pypi.nvidia.com \ cuml-cu11 --extra-index-url=https://pypi.nvidia.com \ cugraph-cu11 --extra-index-url=https://pypi.nvidia.com \ cucim --extra-index-url=https://pypi.nvidia.com
内容的提问来源于stack exchange,提问作者Steven
相关产品推荐
相关产品推荐

