Review comments

.devcontainer/Dockerfile

@@ -31,7 +31,7 @@ ARG USERNAME=triton-server
 RUN apt-get update \
     && apt-get install -y sudo
 
-# RUN pip3 install transformers torch
+RUN pip3 install transformers torch
 
 # Create the user
 RUN apt-get update \

.devcontainer/devcontainer.json

@@ -15,8 +15,9 @@
 			]
 		}
 	},
+	"postCreateCommand": "sudo chown -R triton-server:triton-server ~/.cache",
 
-	"runArgs": [ "--cap-add=SYS_PTRACE", "--security-opt", "seccomp=unconfined", "--gpus=all", "--shm-size=2g", "--ulimit", "stack=67108864"],
+	"runArgs": [ "--cap-add=SYS_PTRACE", "--security-opt", "seccomp=unconfined", "--gpus=all", "--shm-size=2g", "--ulimit", "stack=67108864" ],
 	"mounts": [
 		"source=${localEnv:HOME}/.ssh,target=/home/triton-server/.ssh,type=bind,consistency=cached",
 		"source=${localEnv:HOME}/.cache/huggingface,target=/home/triton-server/.cache/huggingface,type=bind,consistency=cached"

.vscode/tasks.json

@@ -70,7 +70,7 @@
             ]
         },
         {
-            "label": "Build Python",
+            "label": "Build Python Backend",
             "dependsOrder": "sequence",
             "dependsOn": [
                 "Configure",

README.md

@@ -90,6 +90,7 @@ any C++ code.
   - [Custom Metrics](#custom-metrics-1)
 - [Running with Inferentia](#running-with-inferentia)
 - [Logging](#logging)
+- [Development with VSCode](#development-with-vscode)
 - [Reporting problems, asking questions](#reporting-problems-asking-questions)
 
 ## Quick Start
@@ -1810,6 +1811,16 @@ def initialize(self, args):
     # Should print {'custom_key': {'string_value': 'custom_value'}}
 ```
 
+# Development with VSCode
+
+The repository includes a `.devcontainer` folder that contains a `Dockerfile`
+and `devcontainer.json` file to help you develop the Python backend
+using Visual Studio Code.
+
+In order to build the backend, you can the "Build Python Backend" task in the
+[VSCode tasks](https://code.visualstudio.com/docs/editor/tasks). This will build
+the Python backend and install the artifacts in
+`/opt/tritonserver/backends/python`.
 
 
 # Reporting problems, asking questions