{ "cells": [ { "cell_type": "code", "execution_count": 1, "id": "6b269e64-be58-43b5-ad60-0fbd1d37861a", "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "CUDA is available. GPU is ready for use.\n", "Number of GPUs: 1\n", "GPU Name: NVIDIA GeForce RTX 3060\n" ] } ], "source": [ "import torch\n", "\n", "# Check if CUDA is available\n", "if torch.cuda.is_available():\n", " print(\"CUDA is available. GPU is ready for use.\")\n", " print(f\"Number of GPUs: {torch.cuda.device_count()}\")\n", " print(f\"GPU Name: {torch.cuda.get_device_name(0)}\")\n", "else:\n", " print(\"CUDA is not available. No GPU detected.\")\n" ] }, { "cell_type": "code", "execution_count": 2, "id": "97906ea2-b284-4966-9c11-b8629f053815", "metadata": {}, "outputs": [], "source": [ "import plotly" ] }, { "cell_type": "code", "execution_count": null, "id": "53cd2a47-ef77-401a-8497-8fc920919f22", "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "Python 3 (ipykernel)", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.10.13" } }, "nbformat": 4, "nbformat_minor": 5 }