use pkg_resources for PyTorch version checks in notebooks (#191)

* use pkg_resources for PyTorch version checks in notebooks

* remove version checking in notebook code cell

Co-authored-by: Jong Wook Kim <jongwook@nyu.edu>
This commit is contained in:
neverix
2022-04-11 00:08:16 +03:00
committed by GitHub
parent 3482bb6ed3
commit e58d49454c
2 changed files with 5 additions and 7 deletions

View File

@@ -352,10 +352,9 @@
"source": [
"import numpy as np\n",
"import torch\n",
"from pkg_resources import packaging\n",
"\n",
"print(\"Torch version:\", torch.__version__)\n",
"\n",
"assert torch.__version__.split(\".\") >= [\"1\", \"7\", \"1\"], \"PyTorch 1.7.1 or later is required\""
"print(\"Torch version:\", torch.__version__)\n"
],
"execution_count": 2,
"outputs": [

View File

@@ -588,10 +588,9 @@
"import torch\n",
"import clip\n",
"from tqdm.notebook import tqdm\n",
"from pkg_resources import packaging\n",
"\n",
"print(\"Torch version:\", torch.__version__)\n",
"\n",
"assert torch.__version__.split(\".\") >= [\"1\", \"7\", \"1\"], \"PyTorch 1.7.1 or later is required\""
"print(\"Torch version:\", torch.__version__)\n"
],
"execution_count": 2,
"outputs": [
@@ -1105,4 +1104,4 @@
]
}
]
}
}