diff --git a/examples/llmcompressor/workbench_example.ipynb b/examples/llmcompressor/workbench_example.ipynb index 0adbdd22..98f85d16 100644 --- a/examples/llmcompressor/workbench_example.ipynb +++ b/examples/llmcompressor/workbench_example.ipynb @@ -32,7 +32,10 @@ "outputs": [], "source": [ "import torch\n", - "use_gpu = torch.cuda.is_available()" + "use_gpu = torch.cuda.is_available()\n", + "print(f\"GPU acceleration available: {use_gpu}\")\n", + "if not use_gpu:\n", + " print(\"Running in CPU-only mode - operations will be slower and use abbreviated datasets\")" ] }, {