Use image instead of dockerfile in dev container

.devcontainer/devcontainer.json

@@ -1,37 +1,27 @@
-// For format details, see https://aka.ms/devcontainer.json. For config options, see the README at:
-// https://github.com/microsoft/vscode-dev-containers/tree/v0.236.0/containers/docker-existing-dockerfile
 {
 	"name": "Ivy Development Environment",
 
-	// Sets the run context to one level up instead of the .devcontainer folder.
-	"context": "..",
-
-	// Update the 'dockerFile' property if you aren't using the standard 'Dockerfile' filename.
-	// Swap in commented line below if using GPU codespace.
-	// "dockerFile": "../docker/DockerfileGPU",
-
-	"dockerFile": "../docker/Dockerfile",
-
-	// An array of extensions that should be installed into the container.
-	"extensions": ["ms-python.python"],
-
+        // Use unifyai/ivy:latest-gpu for gpu support
+	"image": "unifyai/ivy:latest",
+	"customizations": {
+		"extensions": [
+			"ms-python.python"
+		] 
+	},
 	// Use 'forwardPorts' to make a list of ports inside the container available locally.
 	// "forwardPorts": [],
 
-	// Uncomment the next line to run commands after the container is created - for example installing curl.
 	"postCreateCommand": "bash .devcontainer/post_create_commands.sh"
-
+	
 	// Uncomment when using a ptrace-based debugger like C++, Go, and Rust
 	// "runArgs": [ "--cap-add=SYS_PTRACE", "--security-opt", "seccomp=unconfined" ],
-
+	
 	// Uncomment to use the Docker CLI from inside the container. See https://aka.ms/vscode-remote/samples/docker-from-docker.
 	// "mounts": [ "source=/var/run/docker.sock,target=/var/run/docker.sock,type=bind" ],
-
+	
 	// Uncomment to connect as a non-root user if you've added one. See https://aka.ms/vscode-remote/containers/non-root.
 	// "remoteUser": "vscode"
 
 	// Uncomment if using GPU codespace
 	// "runArgs": ["--gpus","all"],
-
-}
-
+}

.devcontainer/post_create_commands.sh

@@ -1,8 +1,9 @@
 #!/bin/bash
 
-pip3 install black
-pip3 install flake8
-
 git submodule update --init --recursive
 
-python3 -m pip install --user -e .
+python3 -m pip install pre-commit
+
+git config --global --add safe.directory /workspaces/ivy
+
+( cd /workspaces/ivy/ && pre-commit install)