update docker pipfile

openvinotoolkit · Oct 30, 2023 · 4184ff9 · 4184ff9
1 parent d49b666
commit 4184ff9
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 7 deletions.
diff --git a/.docker/Pipfile b/.docker/Pipfile
@@ -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]

diff --git a/.docker/Pipfile.lock b/.docker/Pipfile.lock