diff options
| -rw-r--r-- | train_colab.ipynb | 2 |
1 files changed, 1 insertions, 1 deletions
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": [] }, |