31 lines
689 B
JSON
31 lines
689 B
JSON
{
|
|
"name": "EveryDream2 Dev Container",
|
|
// This Dockerfile requires buildx to be enabled by default
|
|
// `docker buildx install`
|
|
"build": {
|
|
"dockerfile": "../../docker/Dockerfile",
|
|
"target": "runtime"
|
|
},
|
|
"postStartCommand": "/start.sh",
|
|
"forwardPorts": [8888, 6006],
|
|
"containerEnv": {
|
|
"JUPYTER_PASSWORD": "EveryDream"
|
|
},
|
|
|
|
// I don't have a GPU to test this with
|
|
"runArgs": [
|
|
"--gpus",
|
|
"all"
|
|
],
|
|
|
|
// Mimic RunPod/Vast setup
|
|
"workspaceMount": "source=${localWorkspaceFolder},target=/workspace/EveryDream2trainer,type=bind",
|
|
"workspaceFolder": "/workspace/EveryDream2trainer",
|
|
"customizations": {
|
|
"vscode": {
|
|
"extensions": [
|
|
"ms-python.python"
|
|
]
|
|
}
|
|
}
|
|
} |