Skip to content

GitLab

Unverified Commit e864f9c5 authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Graphbolt] Update walkthrough Tutorial. (#6450)

parent a4ccc3e9
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 3 deletions
notebooks/graphbolt/walkthrough.ipynb
View file @ e864f9c5
...@@ -9,7 +9,7 @@ ...@@ -9,7 +9,7 @@
], ],
"gpuType": "T4", "gpuType": "T4",
"private_outputs": true, "private_outputs": true,
"authorship_tag": "ABX9TyPvfttw1smlh4C4S7WfGkTB" "authorship_tag": "ABX9TyOCdFtYQweXnIR1/5oWDSGq"
}, },
"kernelspec": { "kernelspec": {
"name": "python3", "name": "python3",
...@@ -48,7 +48,10 @@ ...@@ -48,7 +48,10 @@
"os.environ['TORCH'] = torch.__version__\n", "os.environ['TORCH'] = torch.__version__\n",
"os.environ['DGLBACKEND'] = \"pytorch\"\n", "os.environ['DGLBACKEND'] = \"pytorch\"\n",
"\n", "\n",
"!pip install --pre dgl -f https://data.dgl.ai/wheels-test/cu118/repo.html > /dev/null\n", "# Uncomment below to install required packages. If the CUDA version is not 11.8,\n",
"# check the https://www.dgl.ai/pages/start.html to find the supported CUDA\n",
"# version and corresponding command to install DGL.\n",
"# !pip install --pre dgl -f https://data.dgl.ai/wheels-test/cu118/repo.html > /dev/null\n",
"\n", "\n",
"try:\n", "try:\n",
" import dgl.graphbolt as gb\n", " import dgl.graphbolt as gb\n",
...@@ -81,7 +84,7 @@ ...@@ -81,7 +84,7 @@
" [4, 9], [4, 9], [5, 9], [9, 9], [5, 9], [9, 9], [7, 9]]\n", " [4, 9], [4, 9], [5, 9], [9, 9], [5, 9], [9, 9], [7, 9]]\n",
")\n", ")\n",
"item_set = gb.ItemSet(node_pairs, names=\"node_pairs\")\n", "item_set = gb.ItemSet(node_pairs, names=\"node_pairs\")\n",
"print(item_set)" "print(list(item_set))"
], ],
"metadata": { "metadata": {
"id": "g73ZAbMQsSgV" "id": "g73ZAbMQsSgV"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment