Skip to content

Commit 41933d8

Browse files
committed
rosettafold: fix torch version
1 parent 6e1d335 commit 41933d8

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

RoseTTAFold.ipynb

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,10 @@
5858
"from IPython.utils import io\n",
5959
"from google.colab import files\n",
6060
"\n",
61+
"import torch\n",
62+
"torch_v = torch.__version__\n",
63+
"\n",
64+
"\n",
6165
"if not os.path.isdir(\"RoseTTAFold\"):\n",
6266
" with io.capture_output() as captured:\n",
6367
" # extra functionality\n",
@@ -81,8 +85,8 @@
8185
"\n",
8286
" # install libraries\n",
8387
" %shell pip install -q dgl-cu113 -f https://data.dgl.ai/wheels/repo.html\n",
84-
" %shell pip install -q torch-scatter -f https://pytorch-geometric.com/whl/torch-1.11.0+cu113.html\n",
85-
" %shell pip install -q torch-sparse -f https://pytorch-geometric.com/whl/torch-1.11.0+cu113.html\n",
88+
" %shell pip install -q torch-scatter -f https://pytorch-geometric.com/whl/torch-{torch_v}.html\n",
89+
" %shell pip install -q torch-sparse -f https://pytorch-geometric.com/whl/torch-{torch_v}.html\n",
8690
" %shell pip install -q torch-geometric\n",
8791
" %shell pip install -q py3Dmol\n",
8892
"\n",

0 commit comments

Comments
 (0)