要在安装PyTorch时避免安装与NVIDIA相关的包,可以使用以下方法:
- 使用CPU版本的PyTorch:PyTorch提供了CPU版本,该版本不需要与NVIDIA相关的包。你可以使用以下命令安装CPU版本的PyTorch:
pip install torch -f https://download.pytorch.org/whl/cpu/torch_stable.html
这将仅安装与CPU兼容的PyTorch版本,而不会安装与NVIDIA相关的包。
- 指定不安装CUDA相关的包:如果你希望安装GPU版本的PyTorch,但不想安装与CUDA(NVIDIA的计算平台)相关的包,可以使用
--no-deps
选项。这将阻止pip安装依赖项,包括与CUDA相关的包。使用以下命令进行安装:
pip install torch --no-deps
请注意,这种方法可能会导致PyTorch在GPU上无法正常工作,因为缺少CUDA支持。
通过以上方法,你可以在安装PyTorch时避免安装与NVIDIA相关的包,从而实现只使用CPU进行推理的需求。
标签:nvidia,PyTorch,pytorch,CUDA,版本,NVIDIA,安装,CPU From: https://blog.51cto.com/M82A1/8133716