如何在安装 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，浪费磁盘空间