jung-geun/PSO
1
0
Fork 0
mirror of https://github.com/jung-geun/PSO.git synced 2025-12-19 20:44:39 +09:00
Code Issues Packages Projects Releases Wiki Activity

dev container 수정 - image 추가(cuda 지원)

Browse Source
This commit is contained in:
jung geun
2024-10-15 09:55:09 +00:00
parent 0883bbe1b6
commit 9745bb7ad4
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
.devcontainer/devcontainer.json
View File

@@ -1,13 +1,9 @@
// For format details, see https://aka.ms/devcontainer.json. For config options, see the // For format details, see https://aka.ms/devcontainer.json. For config options, see the
// README at: https://github.com/devcontainers/templates/tree/main/src/miniconda // README at: https://github.com/devcontainers/templates/tree/main/src/miniconda
{ {
"name": "Miniconda (Python 3)", "image": "nvidia/cuda:12.3.2-runtime-ubuntu22.04",
"name": "PSO with cuda",
// Configure tool-specific properties. // Configure tool-specific properties.
"hostRequirements": {
"cpus": 4,
"memory": "16gb",
"storage": "32gb"
},
"customizations": { "customizations": {
"vscode": { "vscode": {
"extensions": [ "extensions": [
@@ -15,10 +11,15 @@
"ms-toolsai.jupyter", "ms-toolsai.jupyter",
"donjayamanne.python-extension-pack", "donjayamanne.python-extension-pack",
"ms-python.black-formatter", "ms-python.black-formatter",
"github.vscode-github-actions" "github.vscode-github-actions",
"ms-python.isort",
"ms-azuretools.vscode-docker"
] ]
} }
}, },
"runArgs": [
"--gpus=all"
],
// Features to add to the dev container. More info: https://containers.dev/features. // Features to add to the dev container. More info: https://containers.dev/features.
"features": { "features": {
"ghcr.io/devcontainers/features/nvidia-cuda:1": { "ghcr.io/devcontainers/features/nvidia-cuda:1": {
@@ -37,7 +38,8 @@
// "forwardPorts": [], // "forwardPorts": [],
// Use 'postCreateCommand' to run commands after the container is created. // Use 'postCreateCommand' to run commands after the container is created.
"postCreateCommand": [ "postCreateCommand": [
"conda env create --file environment.yaml --name pso" "conda env create --file environment.yaml --name pso",
"nvidia-smi"
] ]
// Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root. // Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
// "remoteUser": "root" // "remoteUser": "root"