如何在安装 pytorch 的时候，不安装 nvidia 相关的包？

  • Z时代
  • 2024-02-10
  • 分类：IT

如何在安装 pytorch 的时候，不安装 nvidia 相关的包？

有些时候，推理只要 cpu，不需要用到 GPU，那么此时，我就不想安装 nvidia 相关的包了

但是 pip install torch 的时候，会把 nvidia 相关的包也一起安装

nvidia-cublas-cu12       12.1.3.1
nvidia-cuda-cupti-cu12   12.1.105
nvidia-cuda-nvrtc-cu12   12.1.105
nvidia-cuda-runtime-cu12 12.1.105
nvidia-cudnn-cu12        8.9.2.26
nvidia-cufft-cu12        11.0.2.54
nvidia-curand-cu12       10.3.2.106
nvidia-cusolver-cu12     11.4.5.107
nvidia-cusparse-cu12     12.1.0.106
nvidia-nccl-cu12         2.18.1
nvidia-nvjitlink-cu12    12.3.52
nvidia-nvtx-cu12         12.1.105

至于为什么会有这个需求，是因为这些 nvidia 包的体积都很巨大，打包出来的 docker image 会多3GB，浪费磁盘空间

回答：

这种问题的想法就是不对的，你得考虑下环境部署的稳定性，有些bug就是因为你前期一开始时的不谨慎导致的。后面再来想办法补救就会很麻烦。

以上是 如何在安装 pytorch 的时候，不安装 nvidia 相关的包？ 的全部内容， 来源链接： utcz.com/p/939051.html

回到顶部