wiklif commited on
Commit
cdc0ee3
·
1 Parent(s): b73fe27

Dockerfile updated

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +6 -6
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-runtime
2
 
3
  WORKDIR /code
4
 
 
1
+ FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
2
 
3
  WORKDIR /code
4
 
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
- torch==1.12.0
2
- torchvision==0.13.0
3
  gradio==3.35.2
4
- Pillow==9.3.0
5
- numpy==1.23.5
6
- matplotlib==3.6.2
7
- scikit-learn==1.1.3
 
1
+ torch==2.0.1
2
+ torchvision==0.15.2
3
  gradio==3.35.2
4
+ Pillow==9.5.0
5
+ numpy==1.24.3
6
+ matplotlib==3.7.1
7
+ scikit-learn==1.2.2