0
点赞
收藏
分享

微信扫一扫

如何在安装 pytorch 的时候,不安装 nvidia 相关的包?

七千22 2023-11-01 阅读 20

要在安装PyTorch时避免安装与NVIDIA相关的包,可以使用以下方法:

  1. 使用CPU版本的PyTorch:PyTorch提供了CPU版本,该版本不需要与NVIDIA相关的包。你可以使用以下命令安装CPU版本的PyTorch:

pip install torch -f https://download.pytorch.org/whl/cpu/torch_stable.html

这将仅安装与CPU兼容的PyTorch版本,而不会安装与NVIDIA相关的包。

  1. 指定不安装CUDA相关的包:如果你希望安装GPU版本的PyTorch,但不想安装与CUDA(NVIDIA的计算平台)相关的包,可以使用--no-deps选项。这将阻止pip安装依赖项,包括与CUDA相关的包。使用以下命令进行安装:

pip install torch --no-deps

请注意,这种方法可能会导致PyTorch在GPU上无法正常工作,因为缺少CUDA支持。

通过以上方法,你可以在安装PyTorch时避免安装与NVIDIA相关的包,从而实现只使用CPU进行推理的需求。

举报

相关推荐

0 条评论