installed pytorch

Dockerfile

@@ -1,5 +1,7 @@
-FROM python:alpine
+FROM python:3.11
 COPY . /app
 WORKDIR /app
 
-CMD python -m http.server 8000
+RUN pip install -r requirements.txt
+
+CMD python3 main.py

main.py

@@ -1 +1,3 @@
-print("Hello World!")
+import torch
+x = torch.rand(5, 3)
+print(x)

requirements.txt

@@ -0,0 +1,16 @@
+certifi==2022.12.7
+charset-normalizer==3.1.0
+filelock==3.10.6
+idna==3.4
+Jinja2==3.1.2
+MarkupSafe==2.1.2
+mpmath==1.3.0
+networkx==3.0
+numpy==1.24.2
+Pillow==9.4.0
+requests==2.28.2
+sympy==1.11.1
+torch==2.0.0
+torchvision==0.15.1
+typing-extensions==4.5.0
+urllib3==1.26.15