import os
os.environ[“CUDA_DEVICE_ORDER”] = “PCI_BUS_ID”
os.environ[“CUDA_VISIBLE_DEVICES”] = “-1”
TensorFlow只在CPU上运行的方法
最新推荐文章于 2025-10-23 17:11:23 发布
本文介绍如何通过设置环境变量来指定CUDA设备的使用顺序及可见设备。这对于多GPU环境或者希望禁用GPU加速的应用场景尤为有用。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
1673





