code-workout-ml/environment-colab-gpu.yml at master · codingalzi/code-workout-ml · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
name: colab312
channels:
  - conda-forge
  - nvidia
  - defaults

dependencies:
  # Core Python
  - python=3.12.12
    # Pip Manager
  - pip

  # Jupyter Environment
  - jupyterlab
  - jupyterlab-myst
  - notebook        # Classic Jupyter Notebook UI
  - ipykernel       # Kernel for Jupyter
  - ipywidgets      # Interactive widgets

  # Documentation
  - jupyter-book

  # Core Data Science
  - numpy
  - pandas
  - pyarrow
  - scipy
  - statsmodels
  - pillow          # Image processing
  - openpyxl        # Excel xlsx
  - xlrd            # Older xls
  - lxml            # HTML/XML parsing

  # Machine Learning
  - scikit-learn
  - lightgbm
  - xgboost
  - catboost
  - flaml
  - optuna

  # Visualization
  - matplotlib
  - seaborn
  - graphviz
  - python-graphviz
  - pydot

  # Utilities
  - requests
  - tqdm

  # Deep Learning Tools
  - tensorboard
  - transformers
  - tokenizers
  - datasets

  # MLOps
  - gradio
  - wandb

  # Interpretability
  - shap

  - pip:
    # Deep Learning Frameworks (CUDA support)
    - keras
    - tensorflow>=2.19.0        # TensorFlow with CUDA (Colab: 2.19.0)
    - protobuf>=6.33.5          # Fix version conflict warning (Latest 6.x)
    - torch>=2.9.0              # PyTorch with CUDA 12.6 (Colab: 2.9.0+cu126)
    - torchvision
    - torchaudio
    - jax[cuda12]>=0.5.3        # JAX with CUDA 12 support
    - jaxlib>=0.5.3

    # Deep Learning Utilities (pip recommended)
    - tensorflow-datasets
    - tensorflow-hub
    - accelerate

    # Kaggle & Data
    - kagglehub                 # Kaggle datasets

    # Advanced ML Tools (pip-only)
    - optuna-dashboard          # Optuna dashboard
    - interpret                 # InterpretML
    - dcor                      # Distance correlation