$:
  vscode:
    - docker:
        image: docker.cnb.cool/examples/language/cuda-12.4
      services:
        - vscode
        - docker
      stages:
        - name: ls
          script: ls

• PyTorch 2.6.0
• CUDA 12.4
• cuDNN 9
• Python 3
• pip
• code-server
• curl
• wget
• git
• vim
• gcc/g++
• cmake
• ninja
• patchelf

• ms-python.python
• ms-python.flake8
• ms-python.black-formatter
• ms-python.mypy-type-checker
• ms-python.pylint
• ms-python.debugpy
• tencent-cloud.coding-copilot

• flash-attn
• ninja
• pyproject
• numpy
• cpufeature
• aiohttp
• zmq
• build
• twine