code-workout-ml/environment-gpu-py312.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
name: gpy312
channels:
  - conda-forge
  - nvidia
  - defaults
dependencies:
  - python=3.12.12
  - pip
  - jupyterlab
  - jupyterlab-myst
  - notebook
  - ipython
  - ipywidgets
  - matplotlib
  - numpy
  - pandas
  - pillow
  - scikit-learn
  - scipy
  - requests
  - tqdm
  - seaborn
  - graphviz
  - python-graphviz
  - pydot
  - flaml
  - catboost
  - openpyxl
  - pip:
    # Latest Keras 3
    - keras
    # TensorFlow with CUDA support (Colab uses 2.19.0)
    - tensorflow[and-cuda]
    # PyTorch with CUDA 12.6 support (Colab uses 2.8.0+cu126)
    - torch --index-url https://download.pytorch.org/whl/cu126
    - torchvision --index-url https://download.pytorch.org/whl/cu126
    - torchaudio --index-url https://download.pytorch.org/whl/cu126
    # JAX with CUDA 12 support (Colab uses 0.5.3)
    - jax[cuda12]>=0.5.3
    - jaxlib>=0.5.3
    # Other useful packages
    - kagglehub
    - tensorflow-datasets
    - tensorflow-hub
    - tensorboard
    - transformers
    - tokenizers
    - datasets
    - accelerate
    - gradio
    - wandb
    - jupyter-book
    # ML libraries
    - lightgbm
    - optuna
    - optuna-dashboard
    - xgboost
    - interpret
    - dcor
    # Interpretability
    - shap