From c433efeadb1100450b3eaab75ecc7767697b644f Mon Sep 17 00:00:00 2001 From: ncoop57 Date: Sun, 18 Jul 2021 14:50:54 +0000 Subject: [PATCH] Update Dockerfile a bit --- .devcontainer/Dockerfile | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile index d9baa36..8736341 100644 --- a/.devcontainer/Dockerfile +++ b/.devcontainer/Dockerfile @@ -1 +1,6 @@ from tensorflow/tensorflow:2.5.0-gpu + +RUN apt update && apt install git vim python3.8 -y +RUN pip install datasets \ + git+https://github.com/huggingface/transformers \ + torch==1.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html \ No newline at end of file