Files
generative-models/.devcontainer/pytorch-1.13+cu117/devcontainer.json
2023-08-12 05:59:57 +00:00

52 lines
1.3 KiB
JSON

{
"name": "Python 3",
"image": "mcr.microsoft.com/devcontainers/python:1-3.8-bullseye",
"features": {
"ghcr.io/devcontainers/features/nvidia-cuda:1": {
"cudaVersion": "11.7"
},
"ghcr.io/devcontainers-contrib/features/hatch:2": {
"version": "latest"
},
"ghcr.io/devcontainers-contrib/features/pipx-package:1": {
"package": "black",
"version": "latest",
"injections": "pylint pytest",
"interpreter": "python3"
},
"ghcr.io/devcontainers-contrib/features/apt-get-packages:1": {
"packages": "libgl1-mesa-glx"
},
"ghcr.io/stability-ai/devcontainer-features/pytorch:1.0.1": {
"version": "1.13.1",
"cudaVersion": "cu117"
}
},
"customizations": {
"vscode": {
"extensions": [
"ms-python.python",
"ms-python.vscode-pylance",
"ms-python.black-formatter"
],
"settings": {
"[python]": {
"diffEditor.ignoreTrimWhitespace": false,
"editor.defaultFormatter": "ms-python.black-formatter",
"editor.formatOnSave": true,
"editor.wordBasedSuggestions": false
},
"python.analysis.typeCheckingMode": "basic",
"black-formatter.args": [
"-l 88"
]
}
}
},
"runArgs": [
"--gpus",
"all"
],
"postCreateCommand": "git config --global --add safe.directory '*'",
"postAttachCommand": "pip install -U -r requirements/pt13.txt;pip install -U -e ."
}