NotRev commited on
Commit
43e7826
·
verified ·
1 Parent(s): e8e9ee9

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,5 +1,5 @@
1
- # Use the official PyTorch base image (CPU-only version)
2
- FROM pytorch/pytorch:2.0.1-cpu
3
 
4
  WORKDIR /app
5
 
 
1
+ # Use the official PyTorch base image with a confirmed tag (CPU-only, Python 3.10)
2
+ FROM pytorch/pytorch:2.0.1-cpu-py310
3
 
4
  WORKDIR /app
5