code-workout-ml/environment-torch-win-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
name: tw-gpu
channels:
  - pytorch
  - nvidia
  - conda-forge
  - 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 # numpy 2.0+ compatibility issues
  - flaml
  - optuna

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

  # Utilities
  - requests
  - tqdm

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

  # MLOps
  - gradio
  - wandb

  # Interpretability
  - shap

  # PyTorch (Windows Native GPU)
  - pytorch::pytorch=*=*cuda*
  - pytorch::torchvision
  - pytorch::torchaudio
  - pytorch::pytorch-cuda=12.4

  - pip:
    # Deep Learning Utilities (pip recommended)
    - protobuf>=6.33.5
    - accelerate

    # Kaggle & Data
    - kagglehub                 # Kaggle datasets

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