Use PyTorch image

.devcontainer/Dockerfile

@@ -1,17 +1,20 @@
-FROM python:3.10-slim
+FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime
+
+ENV DEBIAN_FRONTEND=noninteractive 
 
 WORKDIR /app
 
 COPY . /app/
 
 RUN apt-get update && apt-get install -y \
+    python3-pip \
     build-essential \
     curl \
     software-properties-common \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-RUN pip install -r /app/requirements.txt
+RUN pip3 install -r /app/requirements.txt
 
 EXPOSE 8501
 

.devcontainer/devcontainer.json

@@ -3,6 +3,15 @@
         "dockerfile": "Dockerfile",
         "context": ".."
     },
+    "name": "PyTorch DevContainer",
+    "runArgs": [
+        "--gpus=all",
+        "--shm-size=16gb",
+        "--ipc=host"
+    ],
+    "mounts": [
+        "source=${localEnv:USERPROFILE}/Desktop/workspace/NYU/Artificial Intelligence/cs-gy-6613-assignments/patent_lang_model,target=/app,type=bind,consistency=cached"
+    ],
     "workspaceMount": "source=${localWorkspaceFolder},target=/workspace,type=bind",
     "workspaceFolder": "/workspace",
     "postStartCommand": "echo 'Docker container started...'"