pyproject toml for RL and deep learning on GCP V100 instance with N-series CPU

pyproject.toml

[project]
name = "rl-pytorch-env"
version = "0.1.0"
requires-python = ">=3.10,<3.11"

dependencies = [
  # Core ML
  "torch",
  "torchvision",
  "torchaudio",

  # SciPy stack
  "numpy",
  "scipy",
  "pandas",
  "matplotlib",
  "seaborn",
  "scikit-learn",

  # RL envs
  "gymnasium[classic-control,box2d,atari,accept-rom-license]",

  # RL algorithms
  "stable-baselines3[extra]",

  # HF RL tooling
  "huggingface-hub",
  "huggingface-sb3",
  "datasets",
  "evaluate",

  # Viz / logging
  "tensorboard",
  "wandb",
  "moviepy",
  "pygame",

  # Jupyter
  "jupyterlab",
  "ipywidgets",
]

[tool.uv.pip]
extra-index-url = ["https://download.pytorch.org/whl/cu121"]