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
......@@ -9,7 +9,7 @@
],
"gpuType": "T4",
"private_outputs": true,
"authorship_tag": "ABX9TyPvfttw1smlh4C4S7WfGkTB"
"authorship_tag": "ABX9TyOCdFtYQweXnIR1/5oWDSGq"
},
"kernelspec": {
"name": "python3",
......@@ -48,7 +48,10 @@
"os.environ['TORCH'] = torch.__version__\n",
"os.environ['DGLBACKEND'] = \"pytorch\"\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",
"try:\n",
" import dgl.graphbolt as gb\n",
......@@ -81,7 +84,7 @@
" [4, 9], [4, 9], [5, 9], [9, 9], [5, 9], [9, 9], [7, 9]]\n",
")\n",
"item_set = gb.ItemSet(node_pairs, names=\"node_pairs\")\n",
"print(item_set)"
"print(list(item_set))"
],
"metadata": {
"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