{
"cells": [
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# [Intro](https://github.com/Ziaeemehr/vbi_paper/blob/main/docs/examples/intro.ipynb)\n",
"\n",
"![\"Open](\"https://colab.research.google.com/assets/colab-badge.svg\")
"
]
},
{
"cell_type": "code",
"execution_count": 1,
"metadata": {},
"outputs": [],
"source": [
"import vbi"
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"Integrating: 100%|██████████| 1500/1500 [00:06<00:00, 229.93it/s]\n",
"Integrating: 100%|██████████| 1500/1500 [00:00<00:00, 3116.64it/s]\n",
"Integrating: 100%|██████████| 11999/11999 [00:01<00:00, 6390.68it/s]\n",
"----------------------------------------------------------------------\n",
"Ran 108 tests in 24.049s\n",
"\n",
"OK\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"0.9908645365510211\n"
]
}
],
"source": [
"vbi.tests()"
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
" Dependency Check \n",
" \n",
" Package Version Status \n",
" ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ \n",
" vbi v0.1.3 ✅ Available \n",
" numpy 1.26.4 ✅ Available \n",
" scipy 1.12.0 ✅ Available \n",
" matplotlib 3.10.0 ✅ Available \n",
" sbi 0.23.3 ✅ Available \n",
" torch 2.6.0+cu124 ✅ Available \n",
" cupy 13.3.0 ✅ Available \n",
" \n",
"\n"
],
"text/plain": [
"\u001b[3m Dependency Check \u001b[0m\n",
" \n",
" \u001b[1m \u001b[0m\u001b[1mPackage \u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mVersion \u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mStatus \u001b[0m\u001b[1m \u001b[0m \n",
" ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ \n",
" \u001b[1;36m \u001b[0m\u001b[1;36mvbi \u001b[0m\u001b[1;36m \u001b[0m \u001b[1;32m \u001b[0m\u001b[1;32mv0.1.3 \u001b[0m\u001b[1;32m \u001b[0m \u001b[1;33m \u001b[0m\u001b[1;33m✅ Available\u001b[0m\u001b[1;33m \u001b[0m \n",
" \u001b[1;36m \u001b[0m\u001b[1;36mnumpy \u001b[0m\u001b[1;36m \u001b[0m \u001b[1;32m \u001b[0m\u001b[1;32m1.26.4 \u001b[0m\u001b[1;32m \u001b[0m \u001b[1;33m \u001b[0m\u001b[1;33m✅ Available\u001b[0m\u001b[1;33m \u001b[0m \n",
" \u001b[1;36m \u001b[0m\u001b[1;36mscipy \u001b[0m\u001b[1;36m \u001b[0m \u001b[1;32m \u001b[0m\u001b[1;32m1.12.0 \u001b[0m\u001b[1;32m \u001b[0m \u001b[1;33m \u001b[0m\u001b[1;33m✅ Available\u001b[0m\u001b[1;33m \u001b[0m \n",
" \u001b[1;36m \u001b[0m\u001b[1;36mmatplotlib\u001b[0m\u001b[1;36m \u001b[0m \u001b[1;32m \u001b[0m\u001b[1;32m3.10.0 \u001b[0m\u001b[1;32m \u001b[0m \u001b[1;33m \u001b[0m\u001b[1;33m✅ Available\u001b[0m\u001b[1;33m \u001b[0m \n",
" \u001b[1;36m \u001b[0m\u001b[1;36msbi \u001b[0m\u001b[1;36m \u001b[0m \u001b[1;32m \u001b[0m\u001b[1;32m0.23.3 \u001b[0m\u001b[1;32m \u001b[0m \u001b[1;33m \u001b[0m\u001b[1;33m✅ Available\u001b[0m\u001b[1;33m \u001b[0m \n",
" \u001b[1;36m \u001b[0m\u001b[1;36mtorch \u001b[0m\u001b[1;36m \u001b[0m \u001b[1;32m \u001b[0m\u001b[1;32m2.6.0+cu124\u001b[0m\u001b[1;32m \u001b[0m \u001b[1;33m \u001b[0m\u001b[1;33m✅ Available\u001b[0m\u001b[1;33m \u001b[0m \n",
" \u001b[1;36m \u001b[0m\u001b[1;36mcupy \u001b[0m\u001b[1;36m \u001b[0m \u001b[1;32m \u001b[0m\u001b[1;32m13.3.0 \u001b[0m\u001b[1;32m \u001b[0m \u001b[1;33m \u001b[0m\u001b[1;33m✅ Available\u001b[0m\u001b[1;33m \u001b[0m \n",
" \n"
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"Torch GPU available: True\n",
"\n"
],
"text/plain": [
"\u001b[1;34mTorch GPU available:\u001b[0m \u001b[3;92mTrue\u001b[0m\n"
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"Torch device count: 1\n",
"\n"
],
"text/plain": [
"\u001b[1;34mTorch device count:\u001b[0m \u001b[1;36m1\u001b[0m\n"
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"Torch CUDA version: 12.4\n",
"\n"
],
"text/plain": [
"\u001b[1;34mTorch CUDA version:\u001b[0m \u001b[1;36m12.4\u001b[0m\n"
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"CuPy GPU available: True\n",
"\n"
],
"text/plain": [
"\u001b[1;34mCuPy GPU available:\u001b[0m \u001b[3;92mTrue\u001b[0m\n"
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"CuPy device count: 1\n",
"\n"
],
"text/plain": [
"\u001b[1;34mCuPy device count:\u001b[0m \u001b[1;36m1\u001b[0m\n"
]
},
"metadata": {},
"output_type": "display_data"
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"CUDA Version: 12.6\n",
"Device Name: NVIDIA RTX A5000\n",
"Total Memory: 23.68 GB\n",
"Compute Capability: 8.6\n"
]
}
],
"source": [
"vbi.test_imports()"
]
}
],
"metadata": {
"language_info": {
"name": "python"
}
},
"nbformat": 4,
"nbformat_minor": 2
}