Skip to content

Commit

Permalink
align install pytorch version (#1407)
Browse files Browse the repository at this point in the history
* align install pytorch version

* next part

* update docker pipfile
  • Loading branch information
eaidova authored Oct 30, 2023
1 parent 56daeee commit e6cc928
Show file tree
Hide file tree
Showing 80 changed files with 6,662 additions and 6,033 deletions.
5 changes: 3 additions & 2 deletions .docker/Pipfile
Original file line number Diff line number Diff line change
Expand Up @@ -46,9 +46,10 @@ supervisor = "==4.1.0"
tensorflow = ">=2.5,<=2.12"
tensorflow-datasets = "==4.2.0"
treon = "*"
torch = {index = "pytorch-wheels", version = "==1.13.1+cpu"}
torch = {index = "pytorch-wheels", version = "==2.1.0+cpu"}
torchaudio = {index = "pytorch-wheels", version = "==2.1.0+cpu"}
torchmetrics = ">=0.11.0"
torchvision = {index = "pytorch-wheels", version = "==0.14.1+cpu"}
torchvision = {index = "pytorch-wheels", version = "==0.16.0+cpu"}
transformers = ">=4.21.1"

[dev-packages]
Expand Down
33 changes: 28 additions & 5 deletions .docker/Pipfile.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Original file line number Diff line number Diff line change
Expand Up @@ -853,7 +853,7 @@
],
"metadata": {
"kernelspec": {
"display_name": "Python 3 (ipykernel)",
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
Expand All @@ -867,7 +867,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.10.12"
"version": "3.11.5"
},
"vscode": {
"interpreter": {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -1085,7 +1085,7 @@
],
"metadata": {
"kernelspec": {
"display_name": "Python 3 (ipykernel)",
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
Expand All @@ -1099,7 +1099,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.10.12"
"version": "3.11.5"
}
},
"nbformat": 4,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@
"outputs": [],
"source": [
"%pip install -q \"nncf>=2.5.0\" \n",
"%pip install -q \"transformers[torch]\" datasets evaluate\n",
"%pip install -q \"transformers\" datasets evaluate\n",
"%pip install -q \"openvino>=2023.1.0\""
]
},
Expand Down Expand Up @@ -929,7 +929,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.8.10"
"version": "3.11.5"
},
"vscode": {
"interpreter": {
Expand Down
2 changes: 1 addition & 1 deletion notebooks/106-auto-device/106-auto-device.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -770,7 +770,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.8.10"
"version": "3.11.5"
},
"toc-autonumbering": false,
"toc-showmarkdowntxt": false,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -1067,7 +1067,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.8.10"
"version": "3.11.5"
},
"vscode": {
"interpreter": {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@
"outputs": [],
"source": [
"%pip install -q \"openvino>=2023.1.0\" \"nncf>=2.5.0\"\n",
"%pip install -q soundfile librosa \"transformers[torch]\" datasets \"torchmetrics>=0.11.0\""
"%pip install -q soundfile librosa \"transformers\" datasets \"torchmetrics>=0.11.0\""
]
},
{
Expand Down Expand Up @@ -797,7 +797,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.8.10"
"version": "3.11.5"
},
"vscode": {
"interpreter": {
Expand Down
2 changes: 1 addition & 1 deletion notebooks/108-gpu-device/108-gpu-device.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -1872,7 +1872,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.8.10"
"version": "3.11.5"
},
"vscode": {
"interpreter": {
Expand Down
2 changes: 1 addition & 1 deletion notebooks/109-performance-tricks/109-latency-tricks.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -872,7 +872,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.8.10"
"version": "3.11.5"
},
"widgets": {
"application/vnd.jupyter.widget-state+json": {
Expand Down
Loading

0 comments on commit e6cc928

Please sign in to comment.