From dda6db0777620f8139bd476e27e6b275c0679358 Mon Sep 17 00:00:00 2001 From: haoyuren <13851610112@163.com> Date: Sun, 22 Feb 2026 12:01:17 -0600 Subject: =?UTF-8?q?Fix=20total=5Fmem=20=E2=86=92=20total=5Fmemory=20in=20C?= =?UTF-8?q?olab=20GPU=20check?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-Authored-By: Claude Opus 4.6 --- train_colab.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train_colab.ipynb') diff --git a/train_colab.ipynb b/train_colab.ipynb index a591beb..2a092d7 100644 --- a/train_colab.ipynb +++ b/train_colab.ipynb @@ -34,7 +34,7 @@ { "cell_type": "code", "metadata": {}, - "source": "import torch\nprint(f\"PyTorch: {torch.__version__}\")\nprint(f\"CUDA available: {torch.cuda.is_available()}\")\nif torch.cuda.is_available():\n print(f\"GPU: {torch.cuda.get_device_name(0)}\")\n print(f\"Memory: {torch.cuda.get_device_properties(0).total_mem / 1024**3:.1f} GB\")\nelse:\n print(\"WARNING: No GPU detected. Go to Runtime → Change runtime type → GPU\")", + "source": "import torch\nprint(f\"PyTorch: {torch.__version__}\")\nprint(f\"CUDA available: {torch.cuda.is_available()}\")\nif torch.cuda.is_available():\n print(f\"GPU: {torch.cuda.get_device_name(0)}\")\n print(f\"Memory: {torch.cuda.get_device_properties(0).total_memory / 1024**3:.1f} GB\")\nelse:\n print(\"WARNING: No GPU detected. Go to Runtime → Change runtime type → GPU\")", "execution_count": null, "outputs": [] }, -- cgit v1.2.3