Hugging Face's logo

Spaces:
ejschwartz
/
torch-gpu-demo
Sleeping

App Files Community
ejschwartz commited on
Commit
cd299e4
·
1 Parent(s): 7f27534

change image arch

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -3
Dockerfile CHANGED
@@ -1,6 +1,4 @@
1
- # Base image with CUDA for GPU support
2
- #FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
3
- FROM ubuntu:latest
4
 
5
  ENV PIP_BREAK_SYSTEM_PACKAGES=1
6
 
 
1
+ FROM --platform=linux/amd64 ubuntu:latest
 
 
2
 
3
  ENV PIP_BREAK_SYSTEM_PACKAGES=1
4