Skip to content

Commit 5bcd173

Browse files
Fixes CUDA version issue. (#82)
* fixes CUDA version issue
1 parent 54cc334 commit 5bcd173

File tree

1 file changed

+14
-2
lines changed

1 file changed

+14
-2
lines changed

notebooks/Integrate_ArangoDB_with_PyG.ipynb

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,18 @@
137137
"id": "yeeyjVAVv7Xj"
138138
}
139139
},
140+
{
141+
"cell_type": "code",
142+
"source": [
143+
"!python -c \"import torch; print(torch.__version__)\"\n",
144+
"!python -c \"import torch; print(torch.version.cuda)\""
145+
],
146+
"metadata": {
147+
"id": "4c7pgOdMA2xz"
148+
},
149+
"execution_count": null,
150+
"outputs": []
151+
},
140152
{
141153
"cell_type": "code",
142154
"execution_count": null,
@@ -152,8 +164,8 @@
152164
"!chmod -R 755 ./tools\n",
153165
"!pip install pyarango\n",
154166
"!pip install \"python-arango>=5.0\"\n",
155-
"!pip install -q torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cu113.html\n",
156-
"!pip install -q torch-sparse -f https://data.pyg.org/whl/torch-1.10.0+cu113.html\n",
167+
"!pip install -q torch-scatter -f https://data.pyg.org/whl/torch-1.11.0+cu113.html\n",
168+
"!pip install -q torch-sparse -f https://data.pyg.org/whl/torch-1.11.0+cu113.html\n",
157169
"!pip install -q git+https://github.com/pyg-team/pytorch_geometric.git\n",
158170
"!pip install sentence-transformers"
159171
]

0 commit comments

Comments
 (0)