如何在安装 pytorch 的时候,不安装 nvidia 相关的包?
  VDvlWkTw2thq 2023年11月02日 33 0

要在安装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进行推理的需求。

【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月08日 0

暂无评论

推荐阅读
  wURKzOHw9Irf   2023年12月24日   27   0   0 HTMLicoicohtml
  8l4CZpTOKa7P   2023年12月26日   39   0   0 htmlhtml
  dwHry2iKGG0I   2023年12月26日   31   0   0 githubgithubhtmlhtml
VDvlWkTw2thq