{
  "cells": [
    {
      "cell_type": "markdown",
      "id": "53ef867e-3e0b-4683-b472-145471fe2097",
      "metadata": {
        "id": "53ef867e-3e0b-4683-b472-145471fe2097"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "    \n",
        "# **General Instructions for the ML Coding Problems**\n",
        "\n",
        "Please follow these instructions carefully to ensure a smooth evaluation process.\n",
        "\n",
        "## **1. Notebook Submission**\n",
        "- You **must** make a copy of this notebook and append your **full name** to the filename before submitting (e.g., `[OriginalNotebookName]_[YourName].ipynb`).\n",
        "- Share  your notebook copy with inaio@acmindia.org [This is for your own safety so that you do not accidentally lose any changes while editing the notebook]\n",
        "- After solving the questions, ensure you mention the correct URL of your  modified notebook in the test form\n",
        "- Also answer questions on external resources used and link to LLM chats used for each problem in the main test form\n",
        "\n",
        "## **2. Attempting the Questions**\n",
        "- Carefully **read each problem statement** before attempting.\n",
        "- **Attempt all parts** of each question.\n",
        "- Each question is organized into the following parts\n",
        "   - **DATA**, **TASK**, **HELPER CODE [Optional]** and **ANSWER**\n",
        "- **Follow the function signatures** provided. Do not modify them.\n",
        "- You only need to edit the cells in the **ANSWER** sections\n",
        "- If required, you may also add other modules under **IMPORTS** and **INSTALLATION INSTRUCTIONS**\n",
        "- Do not edit the other cells, especially those marked with **DO NOT MODIFY** which are meant for evaluation\n",
        "- You may add new cells to the notebook with extra code as desired\n",
        "  \n",
        "\n",
        "## **3. Scoring Criteria**\n",
        "Your score will be based on the following factors with distribution varying across each problem.\n",
        "- **Soundness & Creativity** of your approach.  \n",
        "  - Include a clear description and rationale of your solution methodology in the notebook (in markdown cells)\n",
        "  - Solutions that showcase your understanding of data and ML will garner more points\n",
        "- **Code Implementation & Readability**\n",
        "  - Ensure your implementation is correct and works\n",
        "  - Incomplete non-working code will be awarded  partial marks based on problem-wise rubric\n",
        "  - In case you have a solution but are unsure about some aspect, you can define a function that solves that aspect and present the rest of the solution\n",
        "  - Use comments to explain important parts of your code.\n",
        "- **Performance of Your Model**:\n",
        "  - Each task will be assessed based on specified performance metrics both on shared datasets and secret datasets\n",
        "  - Different performance ranges will receive different scores.\n",
        "  - Secret datasets used for last section will be shared along with the final results\n",
        "\n",
        "**Points associated with cells are marked at the beginning of the cell**\n",
        "    \n",
        "## **4. Dataset Usage**\n",
        "- **Only use the datasets provided** in this test.\n",
        "- Do **not** use the provided test data set for training.\n",
        "- Do **not** use external datasets for training or testing.\n",
        "- If the submitted performance metrics cannot be reproduced with your code and original datasets, then you will lose all the points associated with model performance.\n",
        "\n",
        "\n",
        "</div>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "65cb0461-0656-4d25-a67b-3385821ac8d9",
      "metadata": {
        "id": "65cb0461-0656-4d25-a67b-3385821ac8d9"
      },
      "outputs": [],
      "source": []
    },
    {
      "cell_type": "markdown",
      "id": "016204c1-8ed0-4105-8a3d-ea3f7757b05b",
      "metadata": {
        "id": "016204c1-8ed0-4105-8a3d-ea3f7757b05b"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "    \n",
        "## Problem 1:  Analogy Oracle: Cracking the Code of Word Relationships [14 pts]\n",
        "\n",
        "Analogical reasoning is a crucial skill tested in scholastic aptitude exams, where word relationships define logical patterns. Given a pair A:B, the goal is to predict the missing word C:? using learned word relationships. For example:\n",
        "\n",
        "height : tall :: weight : ? (Answer: heavy)\n",
        "cat : kitten :: dog : ? (Answer: puppy)\n",
        "\n",
        "Unfortunately, you are not a native English speaker, but you aim to ML to ace this aptitude test.\n",
        "\n",
        "Your challenge: Developing an AI-powered Analogy Oracle that is as good as an English expert.\n",
        "\n",
        "\n",
        "This problem consists of 4 questions (3 must be attempted, the 4th is private for INAIO evaluation).\n",
        "\n",
        "- **Q1: Zero-Shot Decoding – Solving Analogies Without Training** [5 pts]\n",
        "- **Q2: Train an Analogy Prediction Model** [5 pts]\n",
        "- **Q3: Test Analogy Model on Public Dataset** [2 pts]\n",
        "- **Q4: Test Analogy Model on Private Dataset** [2 pts] [NOT FOR STUDENTS TO ATTEMPT]\n",
        "\n",
        "</div>"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "123c5fbd-9159-42d1-a35c-550c43f68c34",
      "metadata": {
        "id": "123c5fbd-9159-42d1-a35c-550c43f68c34"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "    \n",
        "### INSTALLATION  \n",
        "\n",
        "</div>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "bbe57e50-312a-4a19-82d9-74122c890673",
      "metadata": {
        "colab": {
          "background_save": true,
          "base_uri": "https://localhost:8080/"
        },
        "id": "bbe57e50-312a-4a19-82d9-74122c890673",
        "outputId": "abb5e593-249f-4900-d0af-075d77c75147"
      },
      "outputs": [
        {
          "name": "stdout",
          "output_type": "stream",
          "text": [
            "Collecting uv\n",
            "  Downloading uv-0.6.3-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (11 kB)\n",
            "Downloading uv-0.6.3-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (16.2 MB)\n",
            "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m16.2/16.2 MB\u001b[0m \u001b[31m86.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hInstalling collected packages: uv\n",
            "Successfully installed uv-0.6.3\n",
            "\u001b[2mUsing Python 3.11.11 environment at: /usr\u001b[0m\n",
            "\u001b[2K\u001b[2mResolved \u001b[1m57 packages\u001b[0m \u001b[2min 934ms\u001b[0m\u001b[0m\n",
            "\u001b[2K   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[2K\u001b[1A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2K\u001b[2A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2K\u001b[2A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/197.84 MiB\n",
            "\u001b[2K\u001b[3A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/197.84 MiB\n",
            "\u001b[2K\u001b[3A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/23.50 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/197.84 MiB\n",
            "\u001b[2K\u001b[4A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/53.70 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/197.84 MiB\n",
            "\u001b[2K\u001b[5A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/53.70 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/197.84 MiB\n",
            "\u001b[2K\u001b[6A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/53.70 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/197.84 MiB\n",
            "\u001b[2K\u001b[6A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/53.70 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/197.84 MiB\n",
            "\u001b[2K\u001b[6A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/53.70 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/197.84 MiB\n",
            "\u001b[2K\u001b[6A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/197.84 MiB\n",
            "\u001b[2K\u001b[7A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/13.17 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/197.84 MiB\n",
            "\u001b[2K\u001b[8A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/13.17 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/197.84 MiB\n",
            "\u001b[2K\u001b[8A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/13.17 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/197.84 MiB\n",
            "\u001b[2K\u001b[8A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/13.17 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/197.84 MiB\n",
            "\u001b[2K\u001b[8A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/13.17 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m\u001b[2m------------------------------\u001b[0m\u001b[0m     0 B/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/197.84 MiB\n",
            "\u001b[2K\u001b[8A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/13.17 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 63.92 KiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 127.25 KiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 335.96 KiB/197.84 MiB\n",
            "\u001b[2K\u001b[8A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 223.22 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 76.10 KiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 526.32 KiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 260.54 KiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 495.25 KiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 211.90 KiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 335.96 KiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 16.00 KiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 717.71 KiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 287.22 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 108.10 KiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 702.32 KiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 707.92 KiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 767.25 KiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 639.17 KiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 831.74 KiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 478.45 KiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.14 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 495.12 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 173.29 KiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 1.12 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 883.82 KiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 959.14 KiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 831.17 KiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1007.96 KiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 654.45 KiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.32 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 495.12 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 416.10 KiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 1.12 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 1.12 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.10 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.03 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.23 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 894.45 KiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.57 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 495.12 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 956.10 KiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 1.12 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 1.12 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.10 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.03 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.23 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 894.45 KiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.57 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 527.12 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 1.34 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 1.12 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 1.12 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.31 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.03 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.23 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 894.45 KiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.57 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 527.12 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 1.56 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 1.58 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 1.59 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.31 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.03 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.71 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 894.45 KiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.65 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 546.26 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 1.56 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 1.58 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 1.59 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.31 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.44 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 2.05 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.37 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 2.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 546.26 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 2.06 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 1.98 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 2.12 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 1.83 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.86 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 2.05 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 1.81 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 2.36 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 551.78 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 2.37 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 2.46 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 2.50 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 2.28 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 2.38 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 2.56 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 2.04 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 2.78 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 551.78 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 2.59 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 2.82 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 2.94 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 2.65 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 2.59 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 2.71 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 2.31 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 3.33 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 583.78 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 3.12 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 3.17 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 3.29 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 3.42 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 3.11 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 3.21 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 2.77 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 3.90 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 583.78 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 3.61 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 3.68 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 3.64 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 3.66 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 3.64 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 3.75 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 3.34 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.12 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 583.78 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 3.96 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 3.91 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 4.00 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 3.82 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 3.98 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 3.69 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.44 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 607.22 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 4.21 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 4.03 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 4.19 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 4.20 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 4.18 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.25 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 3.85 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.65 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 607.22 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 4.21 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 4.54 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 4.53 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 4.20 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 4.45 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.25 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.20 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.94 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 607.22 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 4.75 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 4.74 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 4.70 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 4.36 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 4.70 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.44 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.22 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.94 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 607.22 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 4.87 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 5.09 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 4.70 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 4.87 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 4.70 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.75 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.70 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 5.17 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 607.22 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 5.14 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 5.09 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 5.06 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 4.87 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 5.23 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 5.02 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.70 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 5.45 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 607.22 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 5.30 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 5.27 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 5.38 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 5.14 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 5.26 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 5.26 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 4.94 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 5.70 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 607.22 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 5.64 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 5.57 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 5.59 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 5.34 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 5.62 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 5.62 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 5.16 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 6.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-runtime-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 623.22 KiB/863.02 KiB\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 5.83 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 5.79 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 5.80 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 5.66 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 5.79 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 5.78 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 5.46 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 6.23 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[11A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (0/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 6.19 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 6.11 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 6.16 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 5.84 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 6.13 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 6.06 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 5.66 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 6.52 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[10A   \u001b[36m\u001b[1mBuilding\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (1/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 6.19 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 6.11 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 6.16 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 5.84 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 6.13 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 6.06 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 5.66 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 6.52 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[10A      \u001b[32m\u001b[1mBuilt\u001b[0m\u001b[39m wget\u001b[2m==3.2\u001b[0m\n",
            "\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (1/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 6.19 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 6.11 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 6.16 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 5.84 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 6.13 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 6.06 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 5.66 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 6.52 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (1/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 6.39 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 6.36 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 6.38 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 6.31 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 6.30 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 6.37 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 5.97 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 7.14 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (1/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 6.87 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 6.92 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 7.17 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 6.85 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 7.00 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 6.98 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 6.67 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 7.59 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (1/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 7.79 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 7.58 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 7.77 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 7.30 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 7.78 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 7.93 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 7.31 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 8.18 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (1/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 8.18 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 8.36 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 8.42 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 8.02 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 8.28 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 8.43 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 7.89 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 8.70 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (2/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 8.82 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 9.08 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 9.16 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 8.83 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 9.01 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 8.97 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 8.68 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 9.59 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (2/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 9.63 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 9.62 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 9.72 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 9.36 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 9.59 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 9.68 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 9.15 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 10.23 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (2/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 10.14 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 10.15 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 10.21 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 9.84 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 10.11 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 10.15 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 9.87 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-\u001b[2m-----------------------------\u001b[0m\u001b[0m 10.74 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (2/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 10.98 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 10.56 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 10.68 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 10.37 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 10.54 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 11.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 10.27 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 11.57 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (2/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 11.45 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 11.32 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 11.52 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 11.13 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 11.19 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 11.50 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 11.12 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 11.99 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (2/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 11.93 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 11.87 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 12.02 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 11.99 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 11.79 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 12.18 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 11.90 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 12.54 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (2/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 12.52 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 12.36 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 12.62 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 12.53 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 12.55 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 12.73 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 12.13 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 13.24 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (2/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 13.07 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 13.23 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 13.34 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 12.97 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 13.04 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 13.29 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 12.95 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 13.78 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (2/11)\n",
            "\u001b[2mnvidia-cuda-cupti-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 13.17 MiB/13.17 MiB\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 13.78 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 13.89 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 13.63 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 13.60 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 13.79 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 13.57 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 14.48 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[9A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (2/11)\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 14.42 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 14.55 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 14.14 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 14.25 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 14.44 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 14.20 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 15.01 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[8A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (2/11)\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 14.58 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 14.81 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 14.45 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 14.25 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 14.72 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 14.39 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 15.12 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[8A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (2/11)\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 15.28 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 15.41 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 14.94 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 15.05 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 15.37 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 15.03 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 15.94 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[8A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (2/11)\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 15.87 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 16.14 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 15.79 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 15.93 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 16.08 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 15.82 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 16.67 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[8A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (3/11)\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 16.53 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 16.75 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 16.35 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 16.37 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 16.72 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 16.54 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 17.51 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[8A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (3/11)\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 17.35 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 17.48 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 17.27 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 17.24 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 17.48 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 17.28 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 18.34 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[8A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (3/11)\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 18.13 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 18.07 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 18.03 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 17.85 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 18.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 17.90 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 18.95 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[8A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (3/11)\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 18.96 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 18.86 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 18.72 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 18.86 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 18.88 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 18.74 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 19.42 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[8A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (3/11)\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 19.66 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 19.57 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 19.45 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 19.56 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 19.70 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 19.67 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 20.04 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[8A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (3/11)\n",
            "\u001b[2mnvidia-nvjitlink-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 20.09 MiB/20.09 MiB\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 20.49 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 20.48 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 20.31 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 20.50 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 20.51 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 21.32 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[8A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (3/11)\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 21.22 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 21.24 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 21.07 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 21.21 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 21.22 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 21.90 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[7A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (3/11)\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 21.22 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 21.24 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 21.10 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 21.21 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 21.22 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 21.90 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[7A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (3/11)\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 22.29 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 21.82 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 21.56 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 21.78 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 22.23 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--\u001b[2m----------------------------\u001b[0m\u001b[0m 22.75 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[7A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (4/11)\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 22.56 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 22.54 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 22.28 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 22.52 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 22.50 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 23.37 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[7A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (4/11)\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 23.17 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 23.56 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 22.92 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 23.55 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 23.18 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 23.98 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[7A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (4/11)\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 23.48 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 23.98 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 24.12 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 24.34 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 24.20 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 25.10 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[7A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (4/11)\n",
            "\u001b[2mnvidia-cuda-nvrtc-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 23.50 MiB/23.50 MiB\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 25.15 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 24.47 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 25.34 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 24.99 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 26.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[7A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (4/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 25.15 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 24.95 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 25.34 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 24.99 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 26.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 25.87 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 25.67 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 26.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 25.69 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 26.67 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 26.48 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 26.40 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 26.37 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 26.17 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 27.45 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 27.28 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 26.51 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 26.81 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 26.53 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 27.87 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 27.69 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 27.38 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 27.22 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 27.00 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 27.87 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 28.06 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 27.98 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 27.62 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 27.87 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 28.15 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 28.34 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 28.70 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 28.43 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 28.17 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 29.03 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 29.00 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 29.21 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 28.99 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 28.56 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 29.99 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 29.84 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 29.63 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 29.76 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 29.34 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 30.40 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 30.48 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 30.06 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 30.43 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 29.98 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 31.15 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 31.23 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 30.83 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 31.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 30.41 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 31.73 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 31.81 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 31.65 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 31.38 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 30.69 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 32.03 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 32.17 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 31.85 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 31.90 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 31.57 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 32.86 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 32.82 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 32.53 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 32.90 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 31.97 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 33.67 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 33.41 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 33.68 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 33.25 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 32.55 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---\u001b[2m---------------------------\u001b[0m\u001b[0m 34.56 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 34.33 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 33.68 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 34.08 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 33.17 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 34.87 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 34.81 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 34.61 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 34.53 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 33.93 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 35.29 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 35.41 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 34.85 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 34.83 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 34.74 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 35.65 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 35.65 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 35.76 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 35.70 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 35.11 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 36.50 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 36.44 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 36.27 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 36.22 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 35.68 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 37.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 36.76 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 37.04 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 37.12 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 36.02 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 37.68 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 37.86 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 37.75 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 37.76 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 37.17 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 37.98 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 38.86 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 38.66 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 38.53 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 37.90 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 39.03 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 39.53 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 39.58 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 39.31 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 38.34 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 39.83 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 40.56 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 40.25 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 40.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 39.38 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 40.75 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 41.06 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 41.31 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 41.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 40.23 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 41.35 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 41.74 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 42.22 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 41.59 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 41.08 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 42.14 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 42.56 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 43.00 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 42.37 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 41.71 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 43.32 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 43.71 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 43.47 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 43.44 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 42.93 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 44.01 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 44.07 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 44.58 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 43.98 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 43.71 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 44.80 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 45.22 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 45.34 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 45.12 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 44.14 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 45.79 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 45.73 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 45.34 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 45.48 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 44.50 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----\u001b[2m--------------------------\u001b[0m\u001b[0m 46.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 46.05 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 45.85 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 45.92 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 45.40 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 46.58 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 46.05 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 46.27 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 46.02 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 45.73 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 46.89 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 46.60 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 46.55 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 46.90 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 46.19 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 47.37 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 47.37 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 47.47 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 47.25 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 46.59 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 48.18 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 48.26 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 48.22 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 48.07 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 47.24 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 48.53 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 48.86 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 48.95 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 48.77 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 47.96 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 48.98 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 49.60 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 49.59 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 49.23 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 48.75 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 49.84 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 50.40 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 49.91 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 49.70 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 49.12 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 50.59 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 51.03 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 50.95 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 50.48 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 50.04 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 51.25 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 51.50 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 51.40 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 51.25 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 50.87 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 51.81 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 52.31 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 52.39 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 51.62 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 51.80 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 52.62 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 52.65 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 52.73 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 52.37 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 52.12 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 52.93 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 53.17 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 53.43 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 53.32 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 52.86 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 53.48 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 53.66 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 54.10 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 54.24 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 53.45 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 54.54 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 53.70 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 55.06 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 54.72 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 54.45 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 55.26 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-curand-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 53.70 MiB/53.70 MiB\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 55.91 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 55.17 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 55.61 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 55.72 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[6A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 56.32 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 56.07 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 55.93 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 56.72 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (5/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 56.71 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 56.49 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 56.43 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----\u001b[2m-------------------------\u001b[0m\u001b[0m 57.03 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 57.39 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.20 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.00 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 57.79 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 57.39 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.20 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.00 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 57.79 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 57.39 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.20 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.02 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 57.79 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 57.39 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.20 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.02 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 57.79 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 57.42 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.20 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.02 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 57.79 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 57.44 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.20 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.02 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 57.79 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 57.44 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.20 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.02 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 57.79 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 57.44 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.23 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.02 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 57.79 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 57.53 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.33 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 57.61 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 57.93 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 58.41 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 58.18 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 58.50 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 58.90 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 59.37 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 59.18 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 58.87 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 59.66 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 60.31 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 59.98 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 59.42 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 60.29 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 61.00 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 60.78 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 60.27 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 61.08 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 62.06 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 61.60 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 61.29 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 61.69 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 62.48 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 62.75 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 61.75 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 62.65 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 63.55 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 63.10 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 62.62 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 63.47 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 64.39 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 64.21 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 63.35 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 64.44 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 65.04 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 64.84 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 64.34 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 65.12 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 66.00 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 65.72 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 65.36 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 65.45 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 66.43 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 66.18 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 65.62 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 66.37 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 66.98 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 66.18 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 66.05 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 67.02 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 67.42 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 66.83 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 66.44 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 67.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 67.42 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 67.20 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 66.44 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 68.01 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 67.93 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 67.95 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 67.42 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 68.35 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 68.86 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 68.78 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 68.19 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------\u001b[2m------------------------\u001b[0m\u001b[0m 68.67 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 69.18 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 69.53 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 68.64 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 69.42 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 69.83 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 69.66 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 69.94 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 69.97 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 70.67 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 70.73 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 69.94 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 70.92 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 71.18 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 71.16 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 70.94 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 71.23 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 71.98 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 71.90 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 71.56 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 71.80 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 72.78 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 72.90 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 72.01 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 72.17 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 73.13 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 73.36 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 72.94 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 72.87 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 73.68 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 73.71 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 73.55 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 73.11 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 74.28 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 73.83 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 73.58 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 73.71 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 74.44 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 74.96 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 74.50 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 73.92 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 75.76 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 75.32 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 74.93 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 74.14 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 76.01 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 76.14 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 75.65 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 74.44 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 76.76 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 77.05 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 76.36 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 75.17 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 77.16 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 77.45 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 76.90 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 76.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 77.50 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 77.92 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 77.28 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 76.87 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 78.18 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 78.40 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 78.09 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 77.86 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 79.08 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 78.94 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 78.87 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 78.50 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 79.92 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 80.48 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 79.64 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 79.29 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 80.56 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 81.16 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 80.25 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------\u001b[2m-----------------------\u001b[0m\u001b[0m 79.91 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 81.55 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 81.62 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 81.31 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 81.13 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 82.31 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 82.51 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 82.55 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 81.76 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 83.76 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 83.42 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 83.09 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 83.09 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 84.28 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 84.64 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 84.08 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 83.79 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 85.09 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 85.35 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 84.96 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 85.14 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 86.15 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 86.37 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 85.83 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 85.50 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 86.71 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 86.56 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 86.52 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 86.19 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 86.71 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 86.97 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 86.87 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 86.54 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 87.66 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 87.87 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 87.67 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 87.12 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 88.09 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 88.84 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 87.99 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 87.49 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 89.06 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 88.84 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 88.94 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 88.30 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 89.85 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 90.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 89.45 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 88.94 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 90.91 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 90.34 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 89.93 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 89.94 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 91.21 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 91.36 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 90.92 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 90.28 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 91.89 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 91.94 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 91.54 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 90.87 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 92.28 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 92.12 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 91.78 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 90.87 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 92.50 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 93.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 92.56 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 91.32 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 93.23 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 93.44 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 92.83 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 92.35 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 93.34 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 93.44 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 93.51 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------\u001b[2m----------------------\u001b[0m\u001b[0m 92.35 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 93.89 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 93.98 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 93.51 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 92.81 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 94.00 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 94.49 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 93.51 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 93.40 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 94.34 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 95.06 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 94.25 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 93.40 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 95.25 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 95.50 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 94.25 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 94.43 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 95.99 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 96.06 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 95.00 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 95.14 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 96.33 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 96.50 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 95.40 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 95.14 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 96.33 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 96.98 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 96.04 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 95.50 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 97.03 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 96.98 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 96.28 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 96.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 97.37 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 97.42 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 96.28 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 96.90 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 97.92 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 98.17 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 96.97 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 97.05 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 98.24 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 98.53 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 97.31 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 97.43 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 98.24 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 98.53 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 97.81 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 97.95 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 98.87 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 99.75 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 97.81 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 98.29 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 99.21 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 99.75 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 99.20 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 98.29 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 100.23 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 99.75 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 99.29 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 98.29 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 100.23 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 100.56 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 99.29 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 99.09 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 100.93 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 100.56 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 99.29 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 100.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 101.26 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 101.26 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 100.12 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 100.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 101.26 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 101.98 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 100.65 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 100.65 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 102.35 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 102.39 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 100.89 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 101.26 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 102.97 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 102.92 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 101.79 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 101.49 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 103.33 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 103.34 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 102.03 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 101.98 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 103.33 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 103.34 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 102.03 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 102.20 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 103.33 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 103.57 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 102.03 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 102.20 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 103.92 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 103.92 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 102.61 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 102.48 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 104.31 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 104.06 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 103.01 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------\u001b[2m---------------------\u001b[0m\u001b[0m 103.37 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 104.31 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 104.37 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 103.01 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 104.03 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 105.25 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 105.02 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 103.62 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 104.03 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 105.76 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 105.28 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 103.79 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 104.31 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 106.60 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 105.28 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 103.79 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 104.31 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 106.68 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 106.20 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 104.75 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 105.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 106.99 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 106.61 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 105.26 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 105.56 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 107.40 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 107.21 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 105.26 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 106.17 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 107.70 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 107.21 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 105.82 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 106.48 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 108.09 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 107.72 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 106.10 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 106.54 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 108.73 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 108.28 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 106.76 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 106.92 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 109.14 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 108.78 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 107.26 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 107.20 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 109.71 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 108.86 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 107.80 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 107.39 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 109.71 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 109.37 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 107.80 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 107.98 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 110.12 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 109.48 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 108.28 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 107.98 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 110.12 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 110.42 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 108.28 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 108.39 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 110.78 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 110.42 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 108.56 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 108.73 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 111.23 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 110.84 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 108.86 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 109.31 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 111.79 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 111.42 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 109.25 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 109.92 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 112.14 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 111.87 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 109.84 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 110.28 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 112.79 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 112.37 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 110.41 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 110.47 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 113.18 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 112.45 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 110.81 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 111.43 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 113.85 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 113.50 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 110.81 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 111.43 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 114.27 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 113.85 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 111.58 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 112.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 114.27 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 114.47 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 112.30 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 112.25 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 114.77 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 114.47 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 112.56 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 112.25 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 114.77 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 114.47 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 112.56 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 112.92 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 115.12 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 114.91 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 112.97 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 113.44 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 115.12 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 114.91 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 112.97 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 113.81 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 115.12 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 115.24 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 112.97 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 114.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 116.18 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 116.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 112.97 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 114.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 116.20 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 116.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 114.02 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 114.46 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 116.82 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 116.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 114.02 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 114.52 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 116.82 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 116.58 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 115.02 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 115.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 117.88 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 116.94 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 115.02 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------\u001b[2m--------------------\u001b[0m\u001b[0m 115.28 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 118.00 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 117.98 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 115.61 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 116.18 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 118.44 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 118.06 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 116.13 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 116.18 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 119.32 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 118.97 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 116.13 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 116.18 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 119.53 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 119.06 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 117.10 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 117.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 119.53 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 119.36 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 117.10 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 117.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 120.12 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 119.65 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 117.10 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 117.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 120.21 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 119.65 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 117.81 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 117.72 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 120.59 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 119.72 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 118.17 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 118.05 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 121.12 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 119.86 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 118.17 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 118.05 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 121.37 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 119.86 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 119.03 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 118.98 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 121.51 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 120.94 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 119.34 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 119.12 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 121.98 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 121.34 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 119.62 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 119.62 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 122.00 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 121.97 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 120.00 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 119.78 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 122.00 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 122.56 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 120.00 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 120.66 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 122.01 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 122.78 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 120.56 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 121.50 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 122.01 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 123.87 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 121.74 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 121.75 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 122.01 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 124.25 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 122.50 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 122.67 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusolver-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 122.01 MiB/122.01 MiB\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 124.39 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 122.50 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 122.67 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[5A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 124.39 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 122.50 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 122.67 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 124.91 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 123.05 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 122.67 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 125.51 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 123.97 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 124.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (6/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 126.18 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 124.47 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 124.61 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 126.62 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 124.91 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 124.61 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 126.75 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 124.98 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 125.26 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 127.45 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 125.70 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 125.82 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 127.61 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 126.09 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------\u001b[2m-------------------\u001b[0m\u001b[0m 126.04 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 128.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 126.09 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 127.14 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 129.08 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 127.55 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 128.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 129.08 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 128.32 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 128.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 130.20 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 128.32 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 128.97 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 130.20 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 129.58 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 129.78 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 131.65 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 129.97 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 130.16 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 132.14 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 131.19 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 130.20 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 132.33 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 131.77 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 131.44 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 132.70 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 131.77 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 131.44 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 133.85 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 132.44 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 131.44 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 133.85 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 132.80 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 132.57 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 134.23 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 133.04 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 133.29 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 134.73 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 133.04 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 133.56 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 134.94 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 133.88 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 134.09 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 135.76 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 135.00 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 134.93 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 136.67 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 136.11 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 135.98 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 137.50 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 136.78 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 136.65 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 137.89 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 137.41 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 137.18 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 138.72 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 138.17 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 137.92 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 139.62 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 138.17 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------\u001b[2m------------------\u001b[0m\u001b[0m 137.92 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 139.62 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 138.75 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 139.36 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 139.62 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 139.78 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 140.11 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 140.24 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 140.15 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 140.59 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 140.69 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 141.70 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 140.59 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 140.69 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 141.98 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 141.93 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 141.69 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 142.07 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 141.93 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 141.97 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 143.44 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 142.59 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 142.37 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 143.95 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 142.83 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 143.59 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 143.95 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 143.90 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 144.54 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 145.23 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 144.48 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 144.92 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 146.12 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 144.48 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 144.92 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 146.12 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 145.24 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 145.86 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 146.39 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 146.45 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 147.81 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 148.71 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 148.26 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 148.76 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 150.02 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------\u001b[2m-----------------\u001b[0m\u001b[0m 149.36 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 150.39 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 151.12 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 150.72 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 151.63 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 153.29 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 152.32 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 152.86 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 154.69 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 154.58 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 154.41 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 155.92 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 156.61 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 155.18 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 156.87 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 157.71 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 156.42 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 157.98 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 158.57 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 157.39 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 159.21 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------\u001b[2m----------------\u001b[0m\u001b[0m 160.37 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 159.49 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 161.33 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 161.76 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 160.32 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 161.81 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 163.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 161.13 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 162.56 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 163.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 162.30 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 163.68 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 164.51 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 163.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 164.03 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 165.96 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 164.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 165.61 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 166.81 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 164.92 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 166.84 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 168.81 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 166.06 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 167.25 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 169.43 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 166.94 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 168.15 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 170.62 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 168.70 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 169.07 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 171.77 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 169.80 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 170.62 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------\u001b[2m---------------\u001b[0m\u001b[0m 173.25 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 170.32 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 171.15 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 174.26 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 171.47 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 172.62 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 175.23 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 172.24 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 173.20 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 175.95 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 172.72 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 173.81 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 176.65 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 173.48 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 174.56 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 177.39 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 173.94 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 175.34 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 178.77 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 175.08 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 176.50 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 179.91 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 176.23 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 177.59 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 180.68 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 176.90 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 178.00 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 181.74 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 178.00 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 179.09 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 182.63 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 179.12 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 180.34 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------\u001b[2m--------------\u001b[0m\u001b[0m 183.83 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 180.31 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 181.02 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 185.28 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 180.99 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 182.31 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 187.07 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 181.95 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 182.99 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 188.18 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 182.44 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 184.37 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 188.54 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 184.09 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 184.79 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 189.78 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 185.28 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 186.25 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 191.50 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 186.12 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 187.84 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 192.51 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 187.57 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 188.26 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 194.34 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 187.93 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 190.03 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 195.44 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 189.59 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 191.00 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------\u001b[2m-------------\u001b[0m\u001b[0m 196.33 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 190.93 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 192.06 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 198.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 191.89 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 193.78 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 199.10 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 193.09 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 194.85 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 200.47 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 194.40 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 196.13 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 201.61 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 195.31 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 197.87 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 203.54 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 196.89 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 198.90 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 204.44 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 197.83 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 200.01 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------\u001b[2m------------\u001b[0m\u001b[0m 206.16 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 197.83 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 201.62 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 208.14 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 197.84 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cufft-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 201.66 MiB/201.66 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 210.91 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[4A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cusparse-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 197.84 MiB/197.84 MiB\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 212.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[3A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 212.00 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (7/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 213.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 215.59 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 217.70 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------------\u001b[2m-----------\u001b[0m\u001b[0m 218.87 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 220.91 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 222.86 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 224.84 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 226.85 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------------\u001b[2m----------\u001b[0m\u001b[0m 228.86 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 231.42 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 234.11 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 236.05 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 237.91 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------------\u001b[2m---------\u001b[0m\u001b[0m 240.36 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 242.69 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 244.22 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 246.75 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 248.78 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 250.80 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------------\u001b[2m--------\u001b[0m\u001b[0m 252.80 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 254.40 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 256.80 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 258.74 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 260.80 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 261.84 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------------\u001b[2m-------\u001b[0m\u001b[0m 263.93 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 265.87 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 267.50 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 267.50 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 267.59 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 267.61 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 267.62 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 267.64 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 267.65 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 267.67 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 267.69 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 268.07 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 270.33 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 270.33 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 270.33 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 270.33 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 270.39 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.06 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.17 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.19 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.20 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.84 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.84 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.84 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.84 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.84 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.97 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 271.98 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 273.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 273.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 273.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 273.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 273.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 273.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 273.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 273.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 273.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 273.41 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 273.56 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 274.31 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 274.31 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 274.31 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 274.31 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 274.31 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 274.31 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------\u001b[2m------\u001b[0m\u001b[0m 276.34 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 279.07 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 281.89 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 283.91 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 285.91 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-------------------------\u001b[2m-----\u001b[0m\u001b[0m 287.86 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 289.42 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 291.82 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 294.42 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 296.65 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m--------------------------\u001b[2m----\u001b[0m\u001b[0m 298.42 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 301.31 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 304.39 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 306.22 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m---------------------------\u001b[2m---\u001b[0m\u001b[0m 309.71 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 312.42 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 315.71 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 318.47 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m----------------------------\u001b[2m--\u001b[0m\u001b[0m 321.09 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 324.37 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 328.44 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 330.15 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m-----------------------------\u001b[2m-\u001b[0m\u001b[0m 334.34 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 338.31 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 342.43 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2mnvidia-cublas-cu12\u001b[0m \u001b[32m------------------------------\u001b[2m\u001b[0m\u001b[0m 346.40 MiB/346.60 MiB\n",
            "\u001b[2K\u001b[2A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (9/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠦\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠧\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠇\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠋\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠙\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠹\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠸\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠼\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[1A\u001b[37m⠴\u001b[0m \u001b[2mPreparing packages...\u001b[0m (10/11)\n",
            "\u001b[2K\u001b[2mPrepared \u001b[1m11 packages\u001b[0m \u001b[2min 26.11s\u001b[0m\u001b[0m\n",
            "\u001b[2mUninstalled \u001b[1m10 packages\u001b[0m \u001b[2min 38ms\u001b[0m\u001b[0m\n",
            "\u001b[2K\u001b[2mInstalled \u001b[1m11 packages\u001b[0m \u001b[2min 9ms\u001b[0m\u001b[0m\n",
            " \u001b[31m-\u001b[39m \u001b[1mnvidia-cublas-cu12\u001b[0m\u001b[2m==12.5.3.2\u001b[0m\n",
            " \u001b[32m+\u001b[39m \u001b[1mnvidia-cublas-cu12\u001b[0m\u001b[2m==12.4.5.8\u001b[0m\n",
            " \u001b[31m-\u001b[39m \u001b[1mnvidia-cuda-cupti-cu12\u001b[0m\u001b[2m==12.5.82\u001b[0m\n",
            " \u001b[32m+\u001b[39m \u001b[1mnvidia-cuda-cupti-cu12\u001b[0m\u001b[2m==12.4.127\u001b[0m\n",
            " \u001b[31m-\u001b[39m \u001b[1mnvidia-cuda-nvrtc-cu12\u001b[0m\u001b[2m==12.5.82\u001b[0m\n",
            " \u001b[32m+\u001b[39m \u001b[1mnvidia-cuda-nvrtc-cu12\u001b[0m\u001b[2m==12.4.127\u001b[0m\n",
            " \u001b[31m-\u001b[39m \u001b[1mnvidia-cuda-runtime-cu12\u001b[0m\u001b[2m==12.5.82\u001b[0m\n",
            " \u001b[32m+\u001b[39m \u001b[1mnvidia-cuda-runtime-cu12\u001b[0m\u001b[2m==12.4.127\u001b[0m\n",
            " \u001b[31m-\u001b[39m \u001b[1mnvidia-cudnn-cu12\u001b[0m\u001b[2m==9.3.0.75\u001b[0m\n",
            " \u001b[32m+\u001b[39m \u001b[1mnvidia-cudnn-cu12\u001b[0m\u001b[2m==9.1.0.70\u001b[0m\n",
            " \u001b[31m-\u001b[39m \u001b[1mnvidia-cufft-cu12\u001b[0m\u001b[2m==11.2.3.61\u001b[0m\n",
            " \u001b[32m+\u001b[39m \u001b[1mnvidia-cufft-cu12\u001b[0m\u001b[2m==11.2.1.3\u001b[0m\n",
            " \u001b[31m-\u001b[39m \u001b[1mnvidia-curand-cu12\u001b[0m\u001b[2m==10.3.6.82\u001b[0m\n",
            " \u001b[32m+\u001b[39m \u001b[1mnvidia-curand-cu12\u001b[0m\u001b[2m==10.3.5.147\u001b[0m\n",
            " \u001b[31m-\u001b[39m \u001b[1mnvidia-cusolver-cu12\u001b[0m\u001b[2m==11.6.3.83\u001b[0m\n",
            " \u001b[32m+\u001b[39m \u001b[1mnvidia-cusolver-cu12\u001b[0m\u001b[2m==11.6.1.9\u001b[0m\n",
            " \u001b[31m-\u001b[39m \u001b[1mnvidia-cusparse-cu12\u001b[0m\u001b[2m==12.5.1.3\u001b[0m\n",
            " \u001b[32m+\u001b[39m \u001b[1mnvidia-cusparse-cu12\u001b[0m\u001b[2m==12.3.1.170\u001b[0m\n",
            " \u001b[31m-\u001b[39m \u001b[1mnvidia-nvjitlink-cu12\u001b[0m\u001b[2m==12.5.82\u001b[0m\n",
            " \u001b[32m+\u001b[39m \u001b[1mnvidia-nvjitlink-cu12\u001b[0m\u001b[2m==12.4.127\u001b[0m\n",
            " \u001b[32m+\u001b[39m \u001b[1mwget\u001b[0m\u001b[2m==3.2\u001b[0m\n"
          ]
        }
      ],
      "source": [
        "!pip install uv\n",
        "!uv pip install pandas numpy scikit-learn scipy  matplotlib seaborn torch nltk transformers sentence_transformers wget"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "4ac8aa6b-a58f-492a-8541-cd849da6bbae",
      "metadata": {
        "id": "4ac8aa6b-a58f-492a-8541-cd849da6bbae"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "    \n",
        "### IMPORTS\n",
        "</div>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "b2593443-65dd-4769-a76a-8628d5cffdea",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "b2593443-65dd-4769-a76a-8628d5cffdea",
        "outputId": "4ea0b590-2b68-46de-8687-66c9b39002a2"
      },
      "outputs": [
        {
          "name": "stderr",
          "output_type": "stream",
          "text": [
            "[nltk_data] Downloading package words to /root/nltk_data...\n",
            "[nltk_data]   Unzipping corpora/words.zip.\n"
          ]
        }
      ],
      "source": [
        "# EDIT: [O pts]\n",
        "# You may add any other free python packages along with comments\n",
        "\n",
        "# Data Types\n",
        "from typing import Any\n",
        "\n",
        "# Data handling\n",
        "import pandas as pd  # Data manipulation and analysis\n",
        "import numpy as np  # Numerical computations and array handling\n",
        "\n",
        "\n",
        "import nltk\n",
        "nltk.download('words')\n",
        "nltk_vocab = set(nltk.corpus.words.words())  # Use only valid words\n",
        "\n",
        "import os\n",
        "import random\n",
        "from tqdm import tqdm\n",
        "import gensim.downloader as api\n",
        "import gzip\n",
        "import shutil\n",
        "\n",
        "# Machine Learning - Process\n",
        "from sklearn.model_selection import train_test_split  # Splitting dataset\n",
        "from sklearn.pipeline import Pipeline, make_pipeline # Combining multiple processing steps\n",
        "\n",
        "# Machine Learning - Models\n",
        "import torch\n",
        "import torch.nn.functional as F\n",
        "import torch.nn as nn\n",
        "from torch.utils.data import Dataset, DataLoader\n",
        "import gensim\n",
        "from gensim.models import KeyedVectors, Word2Vec\n",
        "from sentence_transformers import SentenceTransformer\n",
        "import sentence_transformers.util as st_utils\n",
        "from transformers import BertModel, BertTokenizer, AdamW\n",
        "\n",
        "\n",
        "# Machine Learning - Feature Transformations\n",
        "from sklearn.preprocessing import OneHotEncoder, StandardScaler # Feature transformations if needed\n",
        "from sklearn.compose import ColumnTransformer #Transforming columns\n",
        "\n",
        "# Model evaluation\n",
        "from sklearn.metrics import (\n",
        "    mean_squared_error, # Mean squared Error\n",
        "    r2_score,  # R² Score\n",
        "    mean_absolute_percentage_error,  # MAPE\n",
        ")\n",
        "from sklearn.metrics.pairwise import cosine_similarity, euclidean_distances\n",
        "\n",
        "# Statistical Analysis\n",
        "from scipy.stats import pearsonr  # Pearson correlation coefficient\n",
        "\n",
        "\n",
        "# Visualization\n",
        "import matplotlib.pyplot as plt  # Plotting graphs\n",
        "import seaborn as sns  # Enhanced data visualization\n"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "YNcXl4JjImef",
      "metadata": {
        "id": "YNcXl4JjImef"
      },
      "source": [
        "### **COPY DATA**"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "G9aOUkJT-5PB",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "G9aOUkJT-5PB",
        "outputId": "1b8c918b-e814-433f-87a9-a7aaf7f54c10"
      },
      "outputs": [
        {
          "name": "stdout",
          "output_type": "stream",
          "text": [
            "--2025-03-02 04:36:27--  https://raw.githubusercontent.com/inaiogit/stage2test/main/test/analogy_test_public.csv\n",
            "Resolving raw.githubusercontent.com (raw.githubusercontent.com)... 185.199.108.133, 185.199.109.133, 185.199.110.133, ...\n",
            "Connecting to raw.githubusercontent.com (raw.githubusercontent.com)|185.199.108.133|:443... connected.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 5130 (5.0K) [text/plain]\n",
            "Saving to: ‘analogy_test_public.csv’\n",
            "\n",
            "\ranalogy_test_public   0%[                    ]       0  --.-KB/s               \ranalogy_test_public 100%[===================>]   5.01K  --.-KB/s    in 0s      \n",
            "\n",
            "2025-03-02 04:36:27 (58.2 MB/s) - ‘analogy_test_public.csv’ saved [5130/5130]\n",
            "\n",
            "--2025-03-02 04:36:28--  https://raw.githubusercontent.com/inaiogit/stage2test/main/test/analogy_train.csv\n",
            "Resolving raw.githubusercontent.com (raw.githubusercontent.com)... 185.199.109.133, 185.199.111.133, 185.199.108.133, ...\n",
            "Connecting to raw.githubusercontent.com (raw.githubusercontent.com)|185.199.109.133|:443... connected.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 7757 (7.6K) [text/plain]\n",
            "Saving to: ‘analogy_train.csv’\n",
            "\n",
            "analogy_train.csv   100%[===================>]   7.58K  --.-KB/s    in 0s      \n",
            "\n",
            "2025-03-02 04:36:28 (76.6 MB/s) - ‘analogy_train.csv’ saved [7757/7757]\n",
            "\n",
            "--2025-03-02 04:36:28--  https://raw.githubusercontent.com/inaiogit/stage2test/main/test/vocab.csv\n",
            "Resolving raw.githubusercontent.com (raw.githubusercontent.com)... 185.199.108.133, 185.199.109.133, 185.199.110.133, ...\n",
            "Connecting to raw.githubusercontent.com (raw.githubusercontent.com)|185.199.108.133|:443... connected.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 4282 (4.2K) [text/plain]\n",
            "Saving to: ‘vocab.csv’\n",
            "\n",
            "vocab.csv           100%[===================>]   4.18K  --.-KB/s    in 0s      \n",
            "\n",
            "2025-03-02 04:36:28 (53.8 MB/s) - ‘vocab.csv’ saved [4282/4282]\n",
            "\n"
          ]
        }
      ],
      "source": [
        "# Copy data\n",
        "!mkdir /content/data\n",
        "!wget https://raw.githubusercontent.com/inaiogit/stage2test/main/test/analogy_test_public.csv\n",
        "!wget https://raw.githubusercontent.com/inaiogit/stage2test/main/test/analogy_train.csv\n",
        "!wget https://raw.githubusercontent.com/inaiogit/stage2test/main/test/vocab.csv\n",
        "!mv analogy_test_public.csv analogy_train.csv vocab.csv data/"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "RC9FohbkkWWP",
      "metadata": {
        "id": "RC9FohbkkWWP"
      },
      "outputs": [],
      "source": [
        "def set_seed(seed):\n",
        "    \"\"\"\n",
        "    Set random seeds for reproducibility.\n",
        "\n",
        "    Args:\n",
        "        seed (int): The seed value to use.\n",
        "    \"\"\"\n",
        "    torch.manual_seed(seed)\n",
        "    torch.cuda.manual_seed_all(seed)\n",
        "    torch.backends.cudnn.deterministic = True\n",
        "    torch.backends.cudnn.benchmark = False\n",
        "    np.random.seed(seed)\n",
        "    random.seed(seed)\n",
        "    os.environ['PYTHONHASHSEED'] = str(seed)\n",
        "\n",
        "seed_value = 42  # Do not change this\n",
        "set_seed(seed_value)"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "9a31fe8d-7d7c-4d0d-85fc-b625fcff025c",
      "metadata": {
        "id": "9a31fe8d-7d7c-4d0d-85fc-b625fcff025c"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "    \n",
        "### **Q1: Zero-Shot Decoding – Solving Analogies Without Training** [2 pts]\n",
        "\n",
        "First task is to solve word analogy problems without training any models. Given a dataset of analogies in the form A:B :: C:?, your goal is to predict the missing word D using only pre-trained word embeddings (such as BERT, Word2Vec, or GloVe). No additional model training is allowed—just smart use of existing embeddings.\n",
        "\n",
        "**Note:** You can look at the full set of (A,B,C) tuples while solving for the missing D's\n",
        "\n",
        "</div>\n"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "f738630c-e2e4-4d33-84da-37b9dd026366",
      "metadata": {
        "id": "f738630c-e2e4-4d33-84da-37b9dd026366"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "\n",
        "### DATA\n",
        "\n",
        "You are provided with an **analogy dataset**\n",
        "\n",
        "- **`analogy_train_path`**: analogy dataset with each row corresponding to two analogous pairs\n",
        "\n",
        "\n",
        "**Columns**\n",
        "- **`A`** (first word in analogy)\n",
        "- **`B`** (second word in analogy, related to A)\n",
        "- **`C`** (third word, forming the analogy with the missing word)\n",
        "- **`D`** (ground truth answer, only for evaluation)\n",
        "\n",
        "\n",
        "</div>\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "109bde31-5c49-4618-a648-5b8e6bb7abc0",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "109bde31-5c49-4618-a648-5b8e6bb7abc0",
        "outputId": "fa61ed4d-5c47-44ff-d8d9-52d40d56914e"
      },
      "outputs": [
        {
          "data": {
            "text/plain": [
              "500"
            ]
          },
          "execution_count": 5,
          "metadata": {},
          "output_type": "execute_result"
        }
      ],
      "source": [
        "# Training datasets\n",
        "analogy_train_path = \"data/analogy_train.csv\"  # analogy data with columns A, B, C, D\n",
        "\n",
        "# Vocabulary\n",
        "# We can use this restricted vocab for the problem to keep inference and training time in check\n",
        "RES_VOCAB = pd.read_csv('data/vocab.csv', header=None)[0].to_list()\n",
        "len(RES_VOCAB)"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "c8a75ea1-4c6b-4959-97a2-97801615682f",
      "metadata": {
        "id": "c8a75ea1-4c6b-4959-97a2-97801615682f"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "    \n",
        "### TASK\n",
        "\n",
        "Analyze the data and record your observations below:\n",
        "   - (a) Perform some exploratory analysis of data and embeddings to come with a solution approach\n",
        "   - (b) Create a function **predict_analogy** as per the signature defined below. If you scroll down, you will see cells with the skeletal code that you need to flesh out.\n",
        "\n",
        "\n",
        "#### **Function 1: `predict_analogy`**\n",
        "\n",
        "```python\n",
        "def predict_analogy(\n",
        "    analogy_df: pd.DataFrame,\n",
        "    model: Any = None,\n",
        "    top_k: int = 5\n",
        ") -> pd.DataFrame:\n",
        "    \"\"\"\n",
        "    Predicts the missing word (D) in an analogy of the form A:B :: C:D using pre-trained embeddings.\n",
        "\n",
        "    Parameters:\n",
        "    - analogy_df (pd.DataFrame): A DataFrame containing:\n",
        "        - \"A\" - First word in analogy\n",
        "        - \"B\" - Second word, related to A\n",
        "        - \"C\" - Third word, forming an analogy with the missing word D\n",
        "    - model (Any, optional): Could be a word embedding model (e.g., Word2Vec, GloVe, or BERT). If None, some hard-coded implementation\n",
        "    - top_k (int, optional): The number of top closest predictions to return. Defaults to 5.\n",
        "\n",
        "    Returns:\n",
        "    - pd.DataFrame: A DataFrame with predictions, containing:\n",
        "        - \"Predicted_D\" - The top predicted word\n",
        "        - \"Top_K_Predictions\" - List of top-k predictions (all lowercase)\n",
        "    \"\"\"\n",
        "```\n",
        "   - (c) Evaluate your strategy using the provided code (no modifications)\n",
        "     \n",
        "</div>\n"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "461d016b-172e-49b1-90e2-a7aa8b1337e1",
      "metadata": {
        "id": "461d016b-172e-49b1-90e2-a7aa8b1337e1"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "    \n",
        "### HELPER CODE\n",
        "</div>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "f3d649a0-e030-4993-883e-af635905136a",
      "metadata": {
        "id": "f3d649a0-e030-4993-883e-af635905136a"
      },
      "outputs": [],
      "source": [
        "glove_vectors = None\n",
        "word2vec_vectors = None\n",
        "def load_embedding_model(model_name=\"word2vec\"):\n",
        "    \"\"\"\n",
        "    Loads a pre-trained word embedding model.\n",
        "\n",
        "    Parameters:\n",
        "    - model_name (str): Name of the model to load. Options: 'word2vec', 'glove', 'bert'\n",
        "\n",
        "    Returns:\n",
        "    - model: Loaded embedding model\n",
        "    \"\"\"\n",
        "    global glove_vectors, word2vec_vectors\n",
        "    if model_name == \"word2vec\":\n",
        "        if word2vec_vectors is None:\n",
        "          print(\"Loading Word2Vec (Google News 300)...\")\n",
        "          word2vec_vectors = api.load(\"word2vec-google-news-300\")\n",
        "        return word2vec_vectors\n",
        "\n",
        "    elif model_name == \"glove\":\n",
        "        if glove_vectors is None:\n",
        "          print(\"Loading GloVe (6B, 300d)...\")\n",
        "          glove_vectors = api.load(\"glove-wiki-gigaword-300\")\n",
        "        return glove_vectors\n",
        "\n",
        "    elif model_name == \"bert\":\n",
        "        print(\"Loading BERT (Sentence Transformers)...\")\n",
        "        return SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')\n",
        "\n",
        "    else:\n",
        "        raise ValueError(\"Unsupported model. Choose from 'word2vec', 'glove', or 'bert'.\")\n",
        "\n",
        "\n",
        "def get_word_embedding(word, model, method=\"word2vec\"):\n",
        "    \"\"\"\n",
        "    Fetches the vector embedding of a word using the specified model.\n",
        "\n",
        "    Parameters:\n",
        "    - word (str): The word to get the embedding for.\n",
        "    - model (Any): Loaded word embedding model.\n",
        "    - method (str): One of 'word2vec', 'glove', 'bert'\n",
        "\n",
        "    Returns:\n",
        "    - np.array: Word embedding vector (or zero vector if word is missing).\n",
        "    \"\"\"\n",
        "    word = word.lower()\n",
        "\n",
        "    if method in [\"word2vec\", \"glove\"]:\n",
        "        return model[word] if word in model else np.zeros(300)\n",
        "\n",
        "    elif method == \"bert\":\n",
        "        return model.encode(word, convert_to_tensor=True)\n",
        "\n",
        "    else:\n",
        "        raise ValueError(\"Unsupported method. Choose from 'word2vec', 'glove', or 'bert'.\")\n",
        "\n",
        "def compute_similarity(vec1, vec2, metric=\"cosine\"):\n",
        "    \"\"\"\n",
        "    Computes similarity between two vectors.\n",
        "\n",
        "    Parameters:\n",
        "    - vec1 (np.array or torch.Tensor): First vector\n",
        "    - vec2 (np.array or torch.Tensor): Second vector\n",
        "    - metric (str): Similarity metric ('cosine' or 'euclidean')\n",
        "\n",
        "    Returns:\n",
        "    - float: Similarity score\n",
        "    \"\"\"\n",
        "    if isinstance(vec1, torch.Tensor):\n",
        "        vec1 = vec1.cpu().numpy()\n",
        "    if isinstance(vec2, torch.Tensor):\n",
        "        vec2 = vec2.cpu().numpy()\n",
        "\n",
        "    vec1 = vec1.reshape(1, -1)\n",
        "    vec2 = vec2.reshape(1, -1)\n",
        "\n",
        "    if metric == \"cosine\":\n",
        "        return cosine_similarity(vec1, vec2)[0][0]\n",
        "\n",
        "    elif metric == \"euclidean\":\n",
        "        return -euclidean_distances(vec1, vec2)[0][0]  # Negative for consistency (higher = better)\n",
        "\n",
        "    else:\n",
        "        raise ValueError(\"Unsupported metric. Choose 'cosine' or 'euclidean'.\")\n",
        "\n",
        "# to save on inference time, we'll encode vocabulary only once for embedding model\n",
        "st_encoded_vocabulary = None\n",
        "def find_closest_words(target_vec, model, method=\"word2vec\", vocab=RES_VOCAB, top_k=5):\n",
        "    \"\"\"\n",
        "    Finds the closest words to a given vector using cosine similarity.\n",
        "\n",
        "    Parameters:\n",
        "    - target_vec (np.array): The vector representation of the target word.\n",
        "    - model (Any): The pre-trained embedding model.\n",
        "    - method (str): Embedding method ('word2vec', 'glove', 'bert')\n",
        "    - vocab (set): Restrict to a vocabulary set (e.g., nltk words, RES_VOCAB)\n",
        "    - top_k (int): Number of top similar words to return\n",
        "\n",
        "    Returns:\n",
        "    - list: Top-k closest words (lowercased)\n",
        "    \"\"\"\n",
        "    global st_encoded_vocabulary\n",
        "    best_matches = []\n",
        "\n",
        "    if method in [\"word2vec\", \"glove\"]:\n",
        "        for word in model.key_to_index:\n",
        "            if vocab and word.lower() not in vocab:\n",
        "                continue  # Skip words not in vocabulary\n",
        "\n",
        "            word_vec = model[word]\n",
        "            similarity = compute_similarity(target_vec, word_vec)\n",
        "            best_matches.append((word, similarity))\n",
        "\n",
        "    elif method == \"bert\":\n",
        "        vocabulary = RES_VOCAB\n",
        "        if st_encoded_vocabulary is None:\n",
        "          print('\\n Encoding the vocab..')\n",
        "          st_encoded_vocabulary = model.encode(list(vocabulary), convert_to_tensor=True)\n",
        "        best_matches = st_utils.semantic_search(target_vec, st_encoded_vocabulary, top_k=top_k)\n",
        "        best_matches = [(list(vocabulary)[match['corpus_id']], match['score']) for match in best_matches[0]]\n",
        "\n",
        "    best_matches = sorted(best_matches, key=lambda x: x[1], reverse=True)[:top_k]\n",
        "    return [word.lower() for word, _ in best_matches]\n",
        "\n",
        "\n",
        "def evaluate_analogy_predictions(\n",
        "    file_path: str,\n",
        "    model: Any = None,\n",
        "    top_k: int = 3\n",
        ") -> dict:\n",
        "    \"\"\"\n",
        "    Evaluates analogy prediction accuracy using Precision@1 and Precision@K from a given file.\n",
        "\n",
        "    Parameters:\n",
        "    - file_path (str): Path to the CSV file containing test analogy data.\n",
        "        Expected columns: \"A\", \"B\", \"C\", \"D\" (ground truth)\n",
        "    - model (Any, optional): If None, we rely on predict_analogy to use the default predictions\n",
        "    - top_k (int, optional): The number of top closest predictions to consider for Precision@K. Defaults to 5.\n",
        "\n",
        "    Returns:\n",
        "    - dict: Dictionary containing:\n",
        "        - \"Precision@1\": Fraction of cases where the top predicted word matches D exactly.\n",
        "        - \"Precision@K\": Fraction of cases where the correct word appears in the top-K predictions.\n",
        "    \"\"\"\n",
        "    # Load test data\n",
        "    try:\n",
        "        test_df = pd.read_csv(file_path)\n",
        "    except Exception as e:\n",
        "        print(f\"Error loading file: {e}\")\n",
        "        return None\n",
        "\n",
        "    # Validate required columns\n",
        "    required_columns = {\"A\", \"B\", \"C\", \"D\"}\n",
        "    if not required_columns.issubset(test_df.columns):\n",
        "        print(f\"Error: Missing required columns. Expected {required_columns}, found {set(test_df.columns)}\")\n",
        "        return None\n",
        "\n",
        "    # Extract only A, B, C columns for prediction\n",
        "    analogy_df = test_df[['A', 'B', 'C']]\n",
        "\n",
        "    # Get predictions (adds \"Predicted_D\" and \"Top_K_Predictions\" columns)\n",
        "    predictions_df = predict_analogy(analogy_df, model=model, top_k=top_k)\n",
        "\n",
        "    # Convert actual D and predicted values to lowercase for case-insensitive comparison\n",
        "    test_df[\"D\"] = test_df[\"D\"].str.lower()\n",
        "    predictions_df[\"Predicted_D\"] = predictions_df[\"Predicted_D\"].str.lower()\n",
        "\n",
        "    # Convert lists of top-K predictions to sets for efficient lookup\n",
        "    predictions_df[\"Top_K_Predictions\"] = predictions_df[\"Top_K_Predictions\"].apply(lambda x: set(map(str.lower, x)))\n",
        "\n",
        "    # Vectorized precision calculations\n",
        "    precision_1 = (predictions_df[\"Predicted_D\"] == test_df[\"D\"]).mean()\n",
        "    precision_k = (predictions_df.apply(lambda row: row[\"Top_K_Predictions\"] and test_df.loc[row.name, \"D\"] in row[\"Top_K_Predictions\"], axis=1)).mean()\n",
        "\n",
        "\n",
        "    return {\n",
        "        \"Precision@1\": precision_1,\n",
        "        \"Precision@K\": precision_k\n",
        "    }"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "e1c7a7db-40a0-4d6c-a1ae-e62a5ea95d9f",
      "metadata": {
        "id": "e1c7a7db-40a0-4d6c-a1ae-e62a5ea95d9f"
      },
      "source": [
        "<div style=\"color:red\">\n",
        "    \n",
        "### ANSWER\n",
        "\n",
        "</div>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "0d984777-5591-4f49-be3b-a9a987ba6116",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 238
        },
        "id": "0d984777-5591-4f49-be3b-a9a987ba6116",
        "outputId": "0ed3e8e9-5e9d-47c6-e126-3906121ebe4b"
      },
      "outputs": [
        {
          "data": {
            "application/vnd.google.colaboratory.intrinsic+json": {
              "summary": "{\n  \"name\": \"data\",\n  \"rows\": 6,\n  \"fields\": [\n    {\n      \"column\": \"A\",\n      \"properties\": {\n        \"dtype\": \"string\",\n        \"num_unique_values\": 6,\n        \"samples\": [\n          \"past\",\n          \"feather\",\n          \"fear\"\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"B\",\n      \"properties\": {\n        \"dtype\": \"string\",\n        \"num_unique_values\": 6,\n        \"samples\": [\n          \"future\",\n          \"light\",\n          \"panic\"\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"C\",\n      \"properties\": {\n        \"dtype\": \"string\",\n        \"num_unique_values\": 6,\n        \"samples\": [\n          \"summer\",\n          \"ice\",\n          \"rain\"\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"D\",\n      \"properties\": {\n        \"dtype\": \"string\",\n        \"num_unique_values\": 6,\n        \"samples\": [\n          \"winter\",\n          \"cold\",\n          \"flood\"\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    }\n  ]\n}",
              "type": "dataframe"
            },
            "text/html": [
              "\n",
              "  <div id=\"df-9723f96a-1599-43f4-a38d-6bf88f4932c4\" class=\"colab-df-container\">\n",
              "    <div>\n",
              "<style scoped>\n",
              "    .dataframe tbody tr th:only-of-type {\n",
              "        vertical-align: middle;\n",
              "    }\n",
              "\n",
              "    .dataframe tbody tr th {\n",
              "        vertical-align: top;\n",
              "    }\n",
              "\n",
              "    .dataframe thead th {\n",
              "        text-align: right;\n",
              "    }\n",
              "</style>\n",
              "<table border=\"1\" class=\"dataframe\">\n",
              "  <thead>\n",
              "    <tr style=\"text-align: right;\">\n",
              "      <th></th>\n",
              "      <th>A</th>\n",
              "      <th>B</th>\n",
              "      <th>C</th>\n",
              "      <th>D</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>248</th>\n",
              "      <td>past</td>\n",
              "      <td>future</td>\n",
              "      <td>summer</td>\n",
              "      <td>winter</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>198</th>\n",
              "      <td>feather</td>\n",
              "      <td>light</td>\n",
              "      <td>ice</td>\n",
              "      <td>cold</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>164</th>\n",
              "      <td>wheel</td>\n",
              "      <td>car</td>\n",
              "      <td>engine</td>\n",
              "      <td>train</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>71</th>\n",
              "      <td>run</td>\n",
              "      <td>quickly</td>\n",
              "      <td>drive</td>\n",
              "      <td>carefully</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>132</th>\n",
              "      <td>light</td>\n",
              "      <td>dark</td>\n",
              "      <td>big</td>\n",
              "      <td>small</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>258</th>\n",
              "      <td>fear</td>\n",
              "      <td>panic</td>\n",
              "      <td>rain</td>\n",
              "      <td>flood</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "</div>\n",
              "    <div class=\"colab-df-buttons\">\n",
              "\n",
              "  <div class=\"colab-df-container\">\n",
              "    <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-9723f96a-1599-43f4-a38d-6bf88f4932c4')\"\n",
              "            title=\"Convert this dataframe to an interactive table.\"\n",
              "            style=\"display:none;\">\n",
              "\n",
              "  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\" viewBox=\"0 -960 960 960\">\n",
              "    <path d=\"M120-120v-720h720v720H120Zm60-500h600v-160H180v160Zm220 220h160v-160H400v160Zm0 220h160v-160H400v160ZM180-400h160v-160H180v160Zm440 0h160v-160H620v160ZM180-180h160v-160H180v160Zm440 0h160v-160H620v160Z\"/>\n",
              "  </svg>\n",
              "    </button>\n",
              "\n",
              "  <style>\n",
              "    .colab-df-container {\n",
              "      display:flex;\n",
              "      gap: 12px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert {\n",
              "      background-color: #E8F0FE;\n",
              "      border: none;\n",
              "      border-radius: 50%;\n",
              "      cursor: pointer;\n",
              "      display: none;\n",
              "      fill: #1967D2;\n",
              "      height: 32px;\n",
              "      padding: 0 0 0 0;\n",
              "      width: 32px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert:hover {\n",
              "      background-color: #E2EBFA;\n",
              "      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "      fill: #174EA6;\n",
              "    }\n",
              "\n",
              "    .colab-df-buttons div {\n",
              "      margin-bottom: 4px;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert {\n",
              "      background-color: #3B4455;\n",
              "      fill: #D2E3FC;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert:hover {\n",
              "      background-color: #434B5C;\n",
              "      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
              "      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
              "      fill: #FFFFFF;\n",
              "    }\n",
              "  </style>\n",
              "\n",
              "    <script>\n",
              "      const buttonEl =\n",
              "        document.querySelector('#df-9723f96a-1599-43f4-a38d-6bf88f4932c4 button.colab-df-convert');\n",
              "      buttonEl.style.display =\n",
              "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "\n",
              "      async function convertToInteractive(key) {\n",
              "        const element = document.querySelector('#df-9723f96a-1599-43f4-a38d-6bf88f4932c4');\n",
              "        const dataTable =\n",
              "          await google.colab.kernel.invokeFunction('convertToInteractive',\n",
              "                                                    [key], {});\n",
              "        if (!dataTable) return;\n",
              "\n",
              "        const docLinkHtml = 'Like what you see? Visit the ' +\n",
              "          '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
              "          + ' to learn more about interactive tables.';\n",
              "        element.innerHTML = '';\n",
              "        dataTable['output_type'] = 'display_data';\n",
              "        await google.colab.output.renderOutput(dataTable, element);\n",
              "        const docLink = document.createElement('div');\n",
              "        docLink.innerHTML = docLinkHtml;\n",
              "        element.appendChild(docLink);\n",
              "      }\n",
              "    </script>\n",
              "  </div>\n",
              "\n",
              "\n",
              "<div id=\"df-81451395-2b2b-45e9-bedc-f856c13e5e78\">\n",
              "  <button class=\"colab-df-quickchart\" onclick=\"quickchart('df-81451395-2b2b-45e9-bedc-f856c13e5e78')\"\n",
              "            title=\"Suggest charts\"\n",
              "            style=\"display:none;\">\n",
              "\n",
              "<svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
              "     width=\"24px\">\n",
              "    <g>\n",
              "        <path d=\"M19 3H5c-1.1 0-2 .9-2 2v14c0 1.1.9 2 2 2h14c1.1 0 2-.9 2-2V5c0-1.1-.9-2-2-2zM9 17H7v-7h2v7zm4 0h-2V7h2v10zm4 0h-2v-4h2v4z\"/>\n",
              "    </g>\n",
              "</svg>\n",
              "  </button>\n",
              "\n",
              "<style>\n",
              "  .colab-df-quickchart {\n",
              "      --bg-color: #E8F0FE;\n",
              "      --fill-color: #1967D2;\n",
              "      --hover-bg-color: #E2EBFA;\n",
              "      --hover-fill-color: #174EA6;\n",
              "      --disabled-fill-color: #AAA;\n",
              "      --disabled-bg-color: #DDD;\n",
              "  }\n",
              "\n",
              "  [theme=dark] .colab-df-quickchart {\n",
              "      --bg-color: #3B4455;\n",
              "      --fill-color: #D2E3FC;\n",
              "      --hover-bg-color: #434B5C;\n",
              "      --hover-fill-color: #FFFFFF;\n",
              "      --disabled-bg-color: #3B4455;\n",
              "      --disabled-fill-color: #666;\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart {\n",
              "    background-color: var(--bg-color);\n",
              "    border: none;\n",
              "    border-radius: 50%;\n",
              "    cursor: pointer;\n",
              "    display: none;\n",
              "    fill: var(--fill-color);\n",
              "    height: 32px;\n",
              "    padding: 0;\n",
              "    width: 32px;\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart:hover {\n",
              "    background-color: var(--hover-bg-color);\n",
              "    box-shadow: 0 1px 2px rgba(60, 64, 67, 0.3), 0 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "    fill: var(--button-hover-fill-color);\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart-complete:disabled,\n",
              "  .colab-df-quickchart-complete:disabled:hover {\n",
              "    background-color: var(--disabled-bg-color);\n",
              "    fill: var(--disabled-fill-color);\n",
              "    box-shadow: none;\n",
              "  }\n",
              "\n",
              "  .colab-df-spinner {\n",
              "    border: 2px solid var(--fill-color);\n",
              "    border-color: transparent;\n",
              "    border-bottom-color: var(--fill-color);\n",
              "    animation:\n",
              "      spin 1s steps(1) infinite;\n",
              "  }\n",
              "\n",
              "  @keyframes spin {\n",
              "    0% {\n",
              "      border-color: transparent;\n",
              "      border-bottom-color: var(--fill-color);\n",
              "      border-left-color: var(--fill-color);\n",
              "    }\n",
              "    20% {\n",
              "      border-color: transparent;\n",
              "      border-left-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "    }\n",
              "    30% {\n",
              "      border-color: transparent;\n",
              "      border-left-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "      border-right-color: var(--fill-color);\n",
              "    }\n",
              "    40% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "    }\n",
              "    60% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "    }\n",
              "    80% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "      border-bottom-color: var(--fill-color);\n",
              "    }\n",
              "    90% {\n",
              "      border-color: transparent;\n",
              "      border-bottom-color: var(--fill-color);\n",
              "    }\n",
              "  }\n",
              "</style>\n",
              "\n",
              "  <script>\n",
              "    async function quickchart(key) {\n",
              "      const quickchartButtonEl =\n",
              "        document.querySelector('#' + key + ' button');\n",
              "      quickchartButtonEl.disabled = true;  // To prevent multiple clicks.\n",
              "      quickchartButtonEl.classList.add('colab-df-spinner');\n",
              "      try {\n",
              "        const charts = await google.colab.kernel.invokeFunction(\n",
              "            'suggestCharts', [key], {});\n",
              "      } catch (error) {\n",
              "        console.error('Error during call to suggestCharts:', error);\n",
              "      }\n",
              "      quickchartButtonEl.classList.remove('colab-df-spinner');\n",
              "      quickchartButtonEl.classList.add('colab-df-quickchart-complete');\n",
              "    }\n",
              "    (() => {\n",
              "      let quickchartButtonEl =\n",
              "        document.querySelector('#df-81451395-2b2b-45e9-bedc-f856c13e5e78 button');\n",
              "      quickchartButtonEl.style.display =\n",
              "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "    })();\n",
              "  </script>\n",
              "</div>\n",
              "\n",
              "    </div>\n",
              "  </div>\n"
            ],
            "text/plain": [
              "           A        B       C          D\n",
              "248     past   future  summer     winter\n",
              "198  feather    light     ice       cold\n",
              "164    wheel      car  engine      train\n",
              "71       run  quickly   drive  carefully\n",
              "132    light     dark     big      small\n",
              "258     fear    panic    rain      flood"
            ]
          },
          "execution_count": 9,
          "metadata": {},
          "output_type": "execute_result"
        }
      ],
      "source": [
        "# EDIT: [0.5 pt]\n",
        "# Add your data and exploration of embeddings code here\n",
        "\n",
        "data = pd.read_csv(analogy_train_path)\n",
        "data.sample(6)"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "df4a5187-08ba-420b-b24a-5c8290e59b8b",
      "metadata": {
        "id": "df4a5187-08ba-420b-b24a-5c8290e59b8b"
      },
      "source": [
        "#### **EDIT: [1 pts]**\n",
        "### **Describe Your Solution Approach**\n",
        "\n",
        "#### **• Data Exploration Notes** [0.5 pt]\n",
        "  -   A and C seem to have no corelation.\n",
        "  -   The corelation between A and B is different in every piece of the data.\n",
        "\n",
        "#### **• Zero Shot Modeling Strategy & Choices** [0.5 pt]\n",
        "  -   I could vectorize A, B, and C and then set D = C+B-A.\n",
        "  -   For finding the top K choices, I could just find the k nearest neighbours to D.\n",
        "  -   I am using a pretrained GloVe model for this.\n",
        "    "
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "6efd5eea-434c-42de-b610-450c3e23e8bf",
      "metadata": {
        "id": "6efd5eea-434c-42de-b610-450c3e23e8bf"
      },
      "outputs": [],
      "source": [
        "# EDIT: [2.5 pts]\n",
        "\n",
        "# Implement the zero-shot prediction of a **predict_analogy** for the case where model = \"None\"\n",
        "# NOTE: For this task, you can choose to  ONLY implement the case model = \"None\" and hardcode your choice\n",
        "# Do not modify the function signature same\n",
        "\n",
        "def predict_analogy(\n",
        "    analogy_df: pd.DataFrame,\n",
        "    model: Any = None,\n",
        "    top_k: int = 5\n",
        ") -> pd.DataFrame:\n",
        "    \"\"\"\n",
        "    Predicts the missing word (D) in an analogy of the form A:B :: C:D using pre-trained embeddings.\n",
        "\n",
        "    Parameters:\n",
        "    - analogy_df (pd.DataFrame): A DataFrame containing:\n",
        "        - \"A\" - First word in analogy\n",
        "        - \"B\" - Second word, related to A\n",
        "        - \"C\" - Third word, forming an analogy with the missing word D\n",
        "    - model (Any, optional): A pre-trained word embedding model (e.g., Word2Vec, GloVe, or BERT). If None, a default Word2Vec model is loaded.\n",
        "    - top_k (int, optional): The number of top closest predictions to return. Defaults to 5.\n",
        "\n",
        "    Returns:\n",
        "    - pd.DataFrame: A DataFrame with predictions, containing:\n",
        "        - \"Predicted_D\" - The top predicted word\n",
        "        - \"Top_K_Predictions\" - List of top-k predictions (all lowercase)\n",
        "    \"\"\"\n",
        "    # Load the pre-trained GloVe model if no model is provided.\n",
        "    glove = api.load(\"glove-wiki-gigaword-300\")\n",
        "\n",
        "    def predict_row(row):\n",
        "        a_word = row[\"A\"]\n",
        "        b_word = row[\"B\"]\n",
        "        c_word = row[\"C\"]\n",
        "\n",
        "        # Retrieve the embeddings if available, otherwise use a zero vector.\n",
        "        A = glove[a_word] if a_word in glove else np.zeros(glove.vector_size)\n",
        "        B = glove[b_word] if b_word in glove else np.zeros(glove.vector_size)\n",
        "        C = glove[c_word] if c_word in glove else np.zeros(glove.vector_size)\n",
        "\n",
        "        # Compute the predicted vector for D: B - A + C.\n",
        "        pred_vector = B - A + C\n",
        "\n",
        "        # Retrieve the top-k most similar words.\n",
        "        predictions = glove.most_similar([pred_vector], topn=top_k)\n",
        "        predicted_d = predictions[0][0]  # Top prediction.\n",
        "        top_k_predictions = [word for word, _ in predictions]\n",
        "\n",
        "        return pd.Series({\n",
        "            \"Predicted_D\": predicted_d,\n",
        "            \"Top_K_Predictions\": top_k_predictions\n",
        "        })\n",
        "\n",
        "    # Apply the prediction row-wise.\n",
        "    results_df = analogy_df.apply(predict_row, axis=1)\n",
        "    return results_df\n",
        "\n",
        "\n",
        "\n",
        "\n",
        "\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "84612fa9-3516-4433-8a52-3ebd28833a09",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "84612fa9-3516-4433-8a52-3ebd28833a09",
        "outputId": "78ee7faa-9c6c-471f-e481-5be9708904e5"
      },
      "outputs": [
        {
          "name": "stdout",
          "output_type": "stream",
          "text": [
            "{'Precision@1': 0.016666666666666666, 'Precision@K': 0.18}\n"
          ]
        }
      ],
      "source": [
        "# DO NOT MODIFY\n",
        "# Run this code to observe the Precision@1 and Precision@3\n",
        "# [pts depend on performance range]\n",
        "print(evaluate_analogy_predictions(analogy_train_path, model=None, top_k=3))"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "0bdf5ea9-bd31-4b62-b624-6daf0b099c5c",
      "metadata": {
        "id": "0bdf5ea9-bd31-4b62-b624-6daf0b099c5c"
      },
      "source": [
        "#### **EDIT: [1 pts]**\n",
        "#### Jot down the performance\n",
        "\n",
        "### **Analogy Performance**\n",
        "  - Precision@1: 0.017\n",
        "  - Precision@3: 0.18\n",
        "\n",
        "### **Any Additional Observations**\n",
        "  -   None"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "90396dfb-9b1b-48e6-8a4e-bff9ad429fc3",
      "metadata": {
        "id": "90396dfb-9b1b-48e6-8a4e-bff9ad429fc3"
      },
      "source": [
        "## **Q2: Train an Analogy Prediction Model** [5 pts]"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "7aa16c19-2d02-4347-9ce4-6a4ac230e0b7",
      "metadata": {
        "id": "7aa16c19-2d02-4347-9ce4-6a4ac230e0b7"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "\n",
        "### DATA\n",
        "\n",
        "Use the same  **analogy dataset** as in Q1\n",
        "\n",
        "- **`analogy_train_path`**: analogy dataset with each row corresponding to two analogous pairs\n",
        "\n",
        "\n",
        "**Columns**\n",
        "- **`A`** (first word in analogy)\n",
        "- **`B`** (second word in analogy, related to A)\n",
        "- **`C`** (third word, forming the analogy with the missing word)\n",
        "- **`D`** (ground truth answer, only for evaluation)\n",
        "\n",
        "</div>\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "acdedf10-2dd4-4aef-ab5a-eab8a681d299",
      "metadata": {
        "id": "acdedf10-2dd4-4aef-ab5a-eab8a681d299"
      },
      "outputs": [],
      "source": [
        "# Training datasets\n",
        "analogy_train_path = \"data/analogy_train.csv\"  # analogy data with columns A, B, C, D"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "5607c19b-2e1a-4fa7-91d0-a3833dce5fc7",
      "metadata": {
        "id": "5607c19b-2e1a-4fa7-91d0-a3833dce5fc7"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "    \n",
        "### TASK\n",
        "\n",
        "Create two functions **learn_analogy_model** and **predict_analogy** as per the signatures defined below.  \n",
        "If you scroll down, you will see cells with the skeletal code that you need to flesh out.\n",
        "\n",
        "---\n",
        "\n",
        "### **Function 1: `learn_analogy_model`**\n",
        "\n",
        "```python\n",
        "def learn_analogy_model(\n",
        "    train_file_path: str\n",
        ") -> Any:\n",
        "    \"\"\"\n",
        "    Loads an analogy dataset from a CSV file, processes it, and trains a machine learning model to predict\n",
        "    the missing word (D) in an analogy of the form A:B :: C:D.\n",
        "\n",
        "    Parameters:\n",
        "    - train_file_path (str): Path to the CSV file containing analogy data with columns:\n",
        "        - \"A\" - First word in analogy\n",
        "        - \"B\" - Second word, related to A\n",
        "        - \"C\" - Third word, forming an analogy with the missing word D\n",
        "        - \"D\" - The ground truth answer (target variable)\n",
        "\n",
        "    Returns:\n",
        "    - model: A trained machine learning model capable of predicting D given A, B, and C.\n",
        "    \"\"\"\n",
        "```\n",
        "---\n",
        "\n",
        "### **Function 2: `predict_analogy`**\n",
        "\n",
        "```python\n",
        "def predict_analogy(\n",
        "    model: Any,\n",
        "    analogy_df: pd.DataFrame,\n",
        "    top_k: int = 3\n",
        ") -> pd.DataFrame:\n",
        "    \"\"\"\n",
        "    Predicts the missing word (D) in an analogy of the form A:B :: C:D using a trained model.\n",
        "\n",
        "    Parameters:\n",
        "    - model (Any): A trained analogy prediction model from `learn_analogy_model`.\n",
        "    - analogy_df (pd.DataFrame): A DataFrame containing:\n",
        "        - \"A\" - First word in analogy\n",
        "        - \"B\" - Second word, related to A\n",
        "        - \"C\" - Third word, forming an analogy with the missing word D\n",
        "    - top_k (int, optional): The number of top closest predictions to return. Defaults to 5.\n",
        "\n",
        "    Returns:\n",
        "    - pd.DataFrame: A DataFrame with predictions, containing:\n",
        "        - \"Predicted_D\" - The top predicted word\n",
        "        - \"Top_K_Predictions\" - List of top-k predictions (all lowercase)\n",
        "    \"\"\"\n",
        "```\n",
        "\n",
        "</div>"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "dd8f6483-e49c-4c8e-acf4-36aafd974782",
      "metadata": {
        "id": "dd8f6483-e49c-4c8e-acf4-36aafd974782"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "    \n",
        "### HELPER CODE\n",
        "</div>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "4aecde05-cf08-4bce-ab76-d2a801331ce1",
      "metadata": {
        "id": "4aecde05-cf08-4bce-ab76-d2a801331ce1"
      },
      "outputs": [],
      "source": [
        "# HELPER CODE\n",
        "# You may choose to use or modify any of the below code in your solution, but it is NOT mandatory\n",
        "\n",
        "def train_word2vec_on_analogy(train_file_path: str, vector_size=300, window=5, min_count=1, epochs=50):\n",
        "    \"\"\"\n",
        "    Trains a Word2Vec model on analogy data to learn custom word embeddings.\n",
        "\n",
        "    Parameters:\n",
        "    - train_file_path (str): Path to the training CSV file with columns A, B, C, D.\n",
        "    - vector_size (int): Dimensionality of word vectors.\n",
        "    - window (int): Maximum distance between current and predicted word in a sentence.\n",
        "    - min_count (int): Minimum count for a word to be included in training.\n",
        "    - epochs (int): Number of training iterations.\n",
        "\n",
        "    Returns:\n",
        "    - model: Trained Word2Vec model.\n",
        "    \"\"\"\n",
        "    df = pd.read_csv(train_file_path)\n",
        "\n",
        "    # Convert analogy pairs into training sentences (treat each analogy as a \"sentence\")\n",
        "    sentences = df[[\"A\", \"B\", \"C\", \"D\"]].values.tolist()\n",
        "\n",
        "    # Train Word2Vec model\n",
        "    model = Word2Vec(sentences, vector_size=vector_size, window=window, min_count=min_count, workers=4, sg=1, epochs=epochs)\n",
        "\n",
        "    # Save trained model\n",
        "    model.save(\"word2vec_analogy.model\")\n",
        "    print(\"Custom Word2Vec model trained and saved!\")\n",
        "\n",
        "    return model\n",
        "\n",
        "def get_trained_word2vec_embedding(\n",
        "    model: KeyedVectors,\n",
        "    word: str,\n",
        "    vector_size: int = 300\n",
        ") -> np.ndarray:\n",
        "    \"\"\"\n",
        "    Retrieves the vector embedding of a word from a trained Word2Vec model.\n",
        "\n",
        "    Parameters:\n",
        "    - model (KeyedVectors): The trained Word2Vec model.\n",
        "    - word (str): The input word to retrieve the embedding for.\n",
        "    - vector_size (int, optional): The size of the word vector. Default is 300.\n",
        "\n",
        "    Returns:\n",
        "    - np.ndarray: The word embedding vector, or a zero vector if the word is not in the model.\n",
        "    \"\"\"\n",
        "    word = word.lower()  # Ensure lowercase lookup\n",
        "\n",
        "    if word in model:\n",
        "        return model[word]\n",
        "    else:\n",
        "        return np.zeros(vector_size)  # Return zero vector if word is not found\n",
        "\n",
        "\n",
        "\n",
        "# Code for training BERT model\n",
        "\n",
        "# Custom Dataset class to handle analogy data\n",
        "class AnalogyDataset(Dataset):\n",
        "    def __init__(self, df, tokenizer):\n",
        "        self.df = df\n",
        "        self.tokenizer = tokenizer\n",
        "\n",
        "    def __len__(self):\n",
        "        return len(self.df)\n",
        "\n",
        "    def __getitem__(self, idx):\n",
        "        # Get the row from the DataFrame\n",
        "        A, B, C, D = self.df.iloc[idx][\"A\"], self.df.iloc[idx][\"B\"], self.df.iloc[idx][\"C\"], self.df.iloc[idx][\"D\"]\n",
        "\n",
        "        # Tokenize the inputs\n",
        "        tokens_A = self.tokenizer(A, return_tensors=\"pt\", padding='max_length', max_length=8)\n",
        "        tokens_B = self.tokenizer(B, return_tensors=\"pt\", padding='max_length', max_length=8)\n",
        "        tokens_C = self.tokenizer(C, return_tensors=\"pt\", padding='max_length', max_length=8)\n",
        "        tokens_D = self.tokenizer(D, return_tensors=\"pt\", padding='max_length', max_length=8)\n",
        "\n",
        "        # The input feature is (A + B - C), and the target is the embedding of D\n",
        "        return tokens_A, tokens_B, tokens_C, tokens_D\n",
        "\n",
        "\n",
        "# BERT-based Model for Analogy Prediction\n",
        "class AnalogyBertModel(nn.Module):\n",
        "    def __init__(self, bert_model):\n",
        "        super(AnalogyBertModel, self).__init__()\n",
        "        self.bert_model = bert_model\n",
        "\n",
        "    def forward(self, tokens_A, tokens_B, tokens_C):\n",
        "        # Pass tokens A, B, and C through the BERT model to get embeddings\n",
        "        output_A = self.bert_model(**tokens_A)  # Output: (last_hidden_state, pooler_output)\n",
        "        output_B = self.bert_model(**tokens_B)\n",
        "        output_C = self.bert_model(**tokens_C)\n",
        "\n",
        "        # Extract the [CLS] token embedding (or alternatively, mean-pooling) as the representation\n",
        "        emb_A = output_A.last_hidden_state[:, 0, :]  # [CLS] token embedding for A\n",
        "        emb_B = output_B.last_hidden_state[:, 0, :]  # [CLS] token embedding for B\n",
        "        emb_C = output_C.last_hidden_state[:, 0, :]  # [CLS] token embedding for C\n",
        "\n",
        "        # The analogy equation: A + B - C\n",
        "        analogy_vector = emb_A + emb_B - emb_C\n",
        "\n",
        "        return analogy_vector\n",
        "\n",
        "\n",
        "\n",
        "def train_bert_on_analogy(train_file_path: str, model_name=\"bert-base-uncased\", epochs=1, batch_size=16):\n",
        "    \"\"\"\n",
        "    Train a BERT model to predict the missing word (D) in analogy tasks.\n",
        "\n",
        "    Parameters:\n",
        "    - train_file_path (str): Path to CSV with analogies (columns: A, B, C, D).\n",
        "    - model_name (str): Pre-trained BERT model to fine-tune.\n",
        "    - epochs (int): Number of training epochs.\n",
        "    - batch_size (int): Batch size for training.\n",
        "\n",
        "    Returns:\n",
        "    - model: Fine-tuned BERT model for analogy prediction.\n",
        "    \"\"\"\n",
        "    # Load tokenizer and BERT model\n",
        "    tokenizer = BertTokenizer.from_pretrained(model_name)\n",
        "    bert_model = BertModel.from_pretrained(model_name)\n",
        "\n",
        "    # Load analogy dataset\n",
        "    df = pd.read_csv(train_file_path)\n",
        "\n",
        "    # Create dataset and dataloader\n",
        "    dataset = AnalogyDataset(df, tokenizer)\n",
        "    dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=True)\n",
        "\n",
        "    # Initialize the analogy model\n",
        "    model = AnalogyBertModel(bert_model)\n",
        "\n",
        "    # Define optimizer and loss function\n",
        "    optimizer = AdamW(model.parameters(), lr=5e-5)\n",
        "    criterion = nn.MSELoss()  # Mean Squared Error for embedding prediction\n",
        "\n",
        "    # Move model to GPU if available\n",
        "    device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
        "    model.to(device)\n",
        "\n",
        "    # Training loop\n",
        "    for epoch in tqdm(range(epochs)):\n",
        "        model.train()\n",
        "        total_loss = 0\n",
        "\n",
        "        for step, (tokens_A, tokens_B, tokens_C, tokens_D) in enumerate(dataloader):\n",
        "            tokens_A = {key: val.squeeze(1).to(device) for key, val in tokens_A.items()}\n",
        "            tokens_B = {key: val.squeeze(1).to(device) for key, val in tokens_B.items()}\n",
        "            tokens_C = {key: val.squeeze(1).to(device) for key, val in tokens_C.items()}\n",
        "            tokens_D = {key: val.squeeze(1).to(device) for key, val in tokens_D.items()}\n",
        "\n",
        "            optimizer.zero_grad()\n",
        "\n",
        "            # Get the predicted embedding\n",
        "            predicted_embedding = model(tokens_A, tokens_B, tokens_C)\n",
        "\n",
        "            # Extract the true embedding for D\n",
        "            true_embedding = model.bert_model(**tokens_D).last_hidden_state[:, 0, :].squeeze()\n",
        "\n",
        "            # Compute the loss (Mean Squared Error between predicted and true embeddings)\n",
        "            loss = criterion(predicted_embedding, true_embedding)\n",
        "\n",
        "            # Backpropagate the loss\n",
        "            loss.backward()\n",
        "            optimizer.step()\n",
        "\n",
        "            total_loss += loss.item()\n",
        "\n",
        "        print(f\"Epoch {epoch + 1}/{epochs}, Loss: {total_loss / len(dataloader)}\")\n",
        "\n",
        "    print(\"Model trained successfully!\")\n",
        "    return model\n",
        "\n",
        "\n",
        "def get_bert_predicted_embedding(model: AnalogyBertModel, tokenizer: BertTokenizer, A: str, B: str, C: str) -> np.ndarray:\n",
        "    \"\"\"\n",
        "    Returns the predicted embedding for a given analogy using the fine-tuned BERT model.\n",
        "\n",
        "    Parameters:\n",
        "    - model: Fine-tuned BERT model for analogy prediction.\n",
        "    - tokenizer: BERT tokenizer.\n",
        "    - A, B, C: Words in the analogy.\n",
        "\n",
        "    Returns:\n",
        "    - np.ndarray: The predicted embedding vector.\n",
        "    \"\"\"\n",
        "    model.eval()\n",
        "    with torch.no_grad():\n",
        "      tokens_A = tokenizer(A, return_tensors=\"pt\", padding='max_length', max_length=8)\n",
        "      tokens_B = tokenizer(B, return_tensors=\"pt\", padding='max_length', max_length=8)\n",
        "      tokens_C = tokenizer(C, return_tensors=\"pt\", padding='max_length', max_length=8)\n",
        "\n",
        "      predicted_embedding = model(tokens_A, tokens_B, tokens_C)\n",
        "      return predicted_embedding.cpu().numpy()\n",
        "\n"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "ebb0a1cf-b311-453b-b33e-b389d46dc8ea",
      "metadata": {
        "id": "ebb0a1cf-b311-453b-b33e-b389d46dc8ea"
      },
      "source": [
        "<div style=\"color:red\">\n",
        "    \n",
        "### ANSWER\n",
        "\n",
        "</div>"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "480dcef1-34ce-4535-8eb1-81ec4eecc4d1",
      "metadata": {
        "id": "480dcef1-34ce-4535-8eb1-81ec4eecc4d1"
      },
      "source": [
        "#### **EDIT: [1.5 pts]**\n",
        "#### You can jot down initial notes here and flesh this out in more detail after the implementation.\n",
        "\n",
        "#### **Describe Your Solution Approach**\n",
        "\n",
        "**Understanding of Available Options:**\n",
        "- Use pre-trained embeddings (GloVe, Word2Vec, or BERT) to capture semantic relationships.\n",
        "- Leverage vector arithmetic (B - A + C) to infer the missing word.\n",
        "- Utilize libraries like Gensim and scikit‑learn for efficient implementation.\n",
        "\n",
        "**Modeling Strategy & Choices:**\n",
        "- **Embedding Selection:** Prefer static embeddings (GloVe/Word2Vec) for simplicity.\n",
        "- **Analogy Calculation:** Compute D = B - A + C.\n",
        "- **Pipeline Design:** Build a scikit‑learn pipeline with a custom transformer for embedding extraction and a classifier or nearest-neighbor search.\n",
        "- **Robustness:** Handle out-of-vocabulary words by substituting with zero vectors.\n",
        "\n",
        "These concise notes capture the key elements of the solution approach."
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "727e0871-fa0e-49e6-b09d-ad8999207ad4",
      "metadata": {
        "id": "727e0871-fa0e-49e6-b09d-ad8999207ad4"
      },
      "outputs": [],
      "source": [
        "# EDIT: [O pts]\n",
        "# Add any additional code that you need for your modeling\n",
        "# Points for any code in this will be assigned to the learn_analogy_model and predict_analogy cells\n",
        "\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "2528f9cb-2248-4849-bfa8-452578bf9129",
      "metadata": {
        "id": "2528f9cb-2248-4849-bfa8-452578bf9129"
      },
      "outputs": [],
      "source": [
        "# @title\n",
        "# EDIT: [2.5 pts]\n",
        "# Implement the training of analogy model\n",
        "# Do not change the signature\n",
        "# Note: just loading a pre-trained model will NOT fetch points - You have to train it on the provided data\n",
        "\n",
        "def learn_analogy_model(\n",
        "    train_file_path: str\n",
        ") -> Any:\n",
        "    \"\"\"\n",
        "    Loads an analogy dataset from a CSV file, processes it, and trains a machine learning model to predict\n",
        "    the missing word (D) in an analogy of the form A:B :: C:D.\n",
        "\n",
        "    Parameters:\n",
        "    - train_file_path (str): Path to the CSV file containing analogy data with columns:\n",
        "        - \"A\" - First word in analogy\n",
        "        - \"B\" - Second word, related to A\n",
        "        - \"C\" - Third word, forming an analogy with the missing word D\n",
        "        - \"D\" - The ground truth answer (target variable)\n",
        "\n",
        "    Returns:\n",
        "    - model: A trained machine learning model capable of predicting D given A, B, and C.\n",
        "    \"\"\"\n",
        "    pass # Your implementation here\n",
        "\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "9e77ecdd-a562-4bf3-ab30-5158f339c906",
      "metadata": {
        "id": "9e77ecdd-a562-4bf3-ab30-5158f339c906"
      },
      "outputs": [],
      "source": [
        "# EDIT: [1 pts]\n",
        "# Implement the prediction of analogy\n",
        "# NOTE: This is similar to Q1 task but we will now run it the model learned from *learn_analogy_model* function\n",
        "# Keep the signature same\n",
        "\n",
        "def predict_analogy(\n",
        "    model: Any,\n",
        "    analogy_df: pd.DataFrame,\n",
        "    top_k: int = 3\n",
        ") -> pd.DataFrame:\n",
        "    \"\"\"\n",
        "    Predicts the missing word (D) in an analogy of the form A:B :: C:D using a trained model.\n",
        "\n",
        "    Parameters:\n",
        "    - model (Any): A trained analogy prediction model from `learn_analogy_model`.\n",
        "    - analogy_df (pd.DataFrame): A DataFrame containing:\n",
        "        - \"A\" - First word in analogy\n",
        "        - \"B\" - Second word, related to A\n",
        "        - \"C\" - Third word, forming an analogy with the missing word D\n",
        "    - top_k (int, optional): The number of top closest predictions to return. Defaults to 3.\n",
        "\n",
        "    Returns:\n",
        "    - pd.DataFrame: A DataFrame with predictions, containing:\n",
        "        - \"Predicted_D\" - The top predicted word\n",
        "        - \"Top_K_Predictions\" - List of top-k predictions (all lowercase)\n",
        "    \"\"\"\n",
        "    pass # Your implementation here\n",
        "\n",
        "\n",
        "\n"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "b2b32973-f713-4c29-9f7e-8af081fab10a",
      "metadata": {
        "id": "b2b32973-f713-4c29-9f7e-8af081fab10a"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "    \n",
        "## **Q3: Test Analogy Model on Public Dataset** [2 pts]\n",
        "\n",
        "</div>"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "174ad98c-1104-4638-af34-13cff87cfe7c",
      "metadata": {
        "id": "174ad98c-1104-4638-af34-13cff87cfe7c"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "\n",
        "### DATA\n",
        "\n",
        "You now get to demonstrate show that you can ace analogies on a unseen test set with the same columns as before\n",
        "\n",
        "- **`analogy_test_path_public`**: analogy dataset with each row corresponding to two analogous pairs\n",
        "\n",
        "\n",
        "**Columns**\n",
        "- **`A`** (first word in analogy)\n",
        "- **`B`** (second word in analogy, related to A)\n",
        "- **`C`** (third word, forming the analogy with the missing word)\n",
        "- **`D`** (ground truth answer, only for evaluation)\n",
        "\n",
        "</div>\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "6545d03d-d186-4e35-a68a-6982dade2017",
      "metadata": {
        "id": "6545d03d-d186-4e35-a68a-6982dade2017"
      },
      "outputs": [],
      "source": [
        "# Public Test Dataset\n",
        "analogy_test_public_path = \"data/analogy_test_public.csv\"  #"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "b5ebae74-5747-43be-bb08-4b751447baa9",
      "metadata": {
        "id": "b5ebae74-5747-43be-bb08-4b751447baa9"
      },
      "source": [
        "<div style=\"color:blue\">  \n",
        "\n",
        "### TASK\n",
        "\n",
        "Execute the code below as is with your implementation of **learn_analogy_model** and **predict_analogy** to test your model\n",
        "\n",
        "- Evaluate your model on this test set.\n",
        "- Compute **Precision@1 and Precision@K**\n",
        "  \n",
        "</div>"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "712424fb-20f3-4e94-ab42-d16b8c8273d2",
      "metadata": {
        "id": "712424fb-20f3-4e94-ab42-d16b8c8273d2"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "    \n",
        "### HELPER CODE\n",
        "</div>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "2b861f33-3a7d-480c-8478-7c16b5294cc3",
      "metadata": {
        "id": "2b861f33-3a7d-480c-8478-7c16b5294cc3"
      },
      "outputs": [],
      "source": [
        "# DO NOT MODIFY\n",
        "# HELPER CODE\n",
        "# This is the same function as in Q1\n",
        "# Use these functions directly since these are meant for evaluation\n",
        "\n",
        "def evaluate_analogy_predictions(\n",
        "    file_path: str,\n",
        "    model: Any = None,\n",
        "    top_k: int = 3\n",
        ") -> dict:\n",
        "    \"\"\"\n",
        "    Evaluates analogy prediction accuracy using Precision@1 and Precision@K from a given file.\n",
        "\n",
        "    Parameters:\n",
        "    - file_path (str): Path to the CSV file containing test analogy data.\n",
        "        Expected columns: \"A\", \"B\", \"C\", \"D\" (ground truth)\n",
        "    - model (Any, optional): If None, we rely on predict_analogy to use the default predictions\n",
        "    - top_k (int, optional): The number of top closest predictions to consider for Precision@K. Defaults to 5.\n",
        "\n",
        "    Returns:\n",
        "    - dict: Dictionary containing:\n",
        "        - \"Precision@1\": Fraction of cases where the top predicted word matches D exactly.\n",
        "        - \"Precision@K\": Fraction of cases where the correct word appears in the top-K predictions.\n",
        "    \"\"\"\n",
        "    # Load test data\n",
        "    try:\n",
        "        test_df = pd.read_csv(file_path)\n",
        "    except Exception as e:\n",
        "        print(f\"Error loading file: {e}\")\n",
        "        return None\n",
        "\n",
        "    # Validate required columns\n",
        "    required_columns = {\"A\", \"B\", \"C\", \"D\"}\n",
        "    if not required_columns.issubset(test_df.columns):\n",
        "        print(f\"Error: Missing required columns. Expected {required_columns}, found {set(test_df.columns)}\")\n",
        "        return None\n",
        "\n",
        "    # Extract only A, B, C columns for prediction\n",
        "    analogy_df = test_df[['A', 'B', 'C']]\n",
        "\n",
        "    # Get predictions (adds \"Predicted_D\" and \"Top_K_Predictions\" columns)\n",
        "    predictions_df = predict_analogy(analogy_df, model=model, top_k=top_k)\n",
        "\n",
        "    # Convert actual D and predicted values to lowercase for case-insensitive comparison\n",
        "    test_df[\"D\"] = test_df[\"D\"].str.lower()\n",
        "    predictions_df[\"Predicted_D\"] = predictions_df[\"Predicted_D\"].str.lower()\n",
        "\n",
        "    # Convert lists of top-K predictions to sets for efficient lookup\n",
        "    predictions_df[\"Top_K_Predictions\"] = predictions_df[\"Top_K_Predictions\"].apply(lambda x: set(map(str.lower, x)))\n",
        "\n",
        "    # Vectorized precision calculations\n",
        "    precision_1 = (predictions_df[\"Predicted_D\"] == test_df[\"D\"]).mean()\n",
        "    precision_k = test_df[\"D\"].isin(predictions_df[\"Top_K_Predictions\"]).mean()\n",
        "\n",
        "    return {\n",
        "        \"Precision@1\": precision_1,\n",
        "        \"Precision@K\": precision_k\n",
        "    }"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "a114f9f5-f1b3-4a1a-ba55-7ce0c573ef08",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 216
        },
        "id": "a114f9f5-f1b3-4a1a-ba55-7ce0c573ef08",
        "outputId": "66c8d571-fcf4-4271-9d46-704d04676d07"
      },
      "outputs": [
        {
          "ename": "NameError",
          "evalue": "name 'learn_analogy_model' is not defined",
          "output_type": "error",
          "traceback": [
            "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
            "\u001b[0;31mNameError\u001b[0m                                 Traceback (most recent call last)",
            "\u001b[0;32m<ipython-input-1-f95465be4b7a>\u001b[0m in \u001b[0;36m<cell line: 0>\u001b[0;34m()\u001b[0m\n\u001b[1;32m      3\u001b[0m \u001b[0;31m# [pts depend on performance range]\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      4\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m----> 5\u001b[0;31m \u001b[0mtrained_model\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mlearn_analogy_model\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0manalogy_train_path\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m      6\u001b[0m \u001b[0mprint\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mevaluate_analogy_predictions\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0manalogy_train_path\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mmodel\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0mtrained_model\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mtop_k\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;36m3\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      7\u001b[0m \u001b[0mprint\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mevaluate_analogy_predictions\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0manalogy_test_public_path\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mmodel\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0mtrained_model\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mtop_k\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;36m3\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
            "\u001b[0;31mNameError\u001b[0m: name 'learn_analogy_model' is not defined"
          ]
        }
      ],
      "source": [
        "# DO NOT MODIFY\n",
        "# Run this code to observe the Precision@1 and Precision@3\n",
        "# [pts depend on performance range]\n",
        "\n",
        "trained_model = learn_analogy_model(analogy_train_path)\n",
        "print(evaluate_analogy_predictions(analogy_train_path, model=trained_model, top_k=3))\n",
        "print(evaluate_analogy_predictions(analogy_test_public_path, model=trained_model, top_k=3))\n",
        "\n"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "06100e12-6da5-42f1-b437-0b17fd92280f",
      "metadata": {
        "id": "06100e12-6da5-42f1-b437-0b17fd92280f"
      },
      "source": [
        "<div style=\"color:red\">\n",
        "    \n",
        "### ANSWER\n",
        "\n",
        "</div>"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "40a1cb92-1a56-4277-b6ad-51dda9dcbd9d",
      "metadata": {
        "id": "40a1cb92-1a56-4277-b6ad-51dda9dcbd9d"
      },
      "source": [
        "#### **EDIT: [2 pts]**\n",
        "#### Jot down the performance\n",
        "\n",
        "### **Train Analogy Performance**\n",
        "  - Precision@1:\n",
        "  - Precision@3:\n",
        "\n",
        "### **Test Analogy Performance**\n",
        "  - Precision@1:\n",
        "  - Precision@3:\n",
        "\n",
        "\n",
        "### **Any Additional Observations**\n",
        "  -   \n",
        "  -\n",
        "\n"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "15e0f606-9220-4647-9165-2077c2c9edc5",
      "metadata": {
        "id": "15e0f606-9220-4647-9165-2077c2c9edc5"
      },
      "source": [
        "<div style=\"color:red\">\n",
        "\n",
        "## YOU CAN STOP THE TEST HERE -- BELOW EVALUATION TO BE PERFORMED BY INAIO\n",
        "\n",
        "</div>"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "55913a7f-d68b-4825-9394-26746d8c9c8b",
      "metadata": {
        "id": "55913a7f-d68b-4825-9394-26746d8c9c8b"
      },
      "source": [
        "<div style=\"color:blue\">\n",
        "\n",
        "## **Q4: Test Analogy Model on Private Dataset** [2 pts]\n",
        "\n",
        "- Same metrics and lead time as Public Dataset\n",
        "\n",
        "</div>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "id": "ea8e6016-4b27-4adb-9f6e-e4481b0c126c",
      "metadata": {
        "id": "ea8e6016-4b27-4adb-9f6e-e4481b0c126c"
      },
      "outputs": [],
      "source": []
    }
  ],
  "metadata": {
    "colab": {
      "provenance": [],
      "toc_visible": true
    },
    "kernelspec": {
      "display_name": "Python 3",
      "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.9.18"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 5
}