当前位置：首页 » AI答疑

如何验证GPU加速功能是否成功启用？

2025-08-19

158

可通过以下步骤验证GPU加速状态：

确保已按NVIDIA GPU方式安装依赖：pip install onnxruntime-gpu和对应版本的PyTorch
在激活的虚拟环境中运行检测命令：
python -c "import torch; print(f'CUDA available: {torch.cuda.is_available()}')"
当输出CUDA available: True时表示配置成功
实际运行时可打开任务管理器(Windows)或使用nvidia-smi命令(Linux)观察GPU利用率

注意事项：

快速查询站内AI工具