在 python 命令前加上 CUDA_VISIBLE_DEVICES 可以指定 GPU

CUDA_VISIBLE_DEVICES=1
CUDA_VISIBLE_DEVICES=0,1,2
CUDA_VISIBLE_DEVICES=2,1,0
Logo
NVIDIA DRIVE 智能汽车专区

更多推荐

cover

NVIDIA DRIVE 合作伙伴在 CES 上展示最新移动出行创新技术

avatar NVIDIA DRIVE 智能汽车专区
cover

NVIDIA DRIVE Hyperion 平台为自动驾驶汽车开发实现关键汽车安全和网络安全里程碑

avatar NVIDIA DRIVE 智能汽车专区

丰田、Aurora 和大陆集团加入 NVIDIA 合作伙伴行列，推出下一代高度自动化的自动驾驶车型

使用或采用 NVIDIA 产品和技术的第三方、这样做所带来的优势和影响以及第三方产品的功能、性能和供货情况；我们依靠第三方来制造、组装、包装和测试我们的产品；NVIDIA、NVIDIA 徽标、NVIDIA Cosmos、NVIDIA DGX、NVIDIA DRIVE、NVIDIA DRIVE AGX、NVIDIA DRIVE AGX Orin、NVIDIA Omniverse 和 NVIDIA O

avatar NVIDIA DRIVE 智能汽车专区