From eff69b13ecfad1ff5e825bdb11ebf7bf6a5e7030 Mon Sep 17 00:00:00 2001
From: William Brannon <wbrannon@media.mit.edu>
Date: Tue, 24 Sep 2024 05:06:33 +0000
Subject: [PATCH] conda env setup script for development

---
 conda-env.sh | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 conda-env.sh

diff --git a/conda-env.sh b/conda-env.sh
new file mode 100644
index 0000000..88d5486
--- /dev/null
+++ b/conda-env.sh
@@ -0,0 +1,25 @@
+#!/bin/bash
+
+# Set up the packages needed for all images in a conda environment, for testing
+
+set -Eeuo pipefail
+set -x
+
+PYTHON_VERSION=3.10
+CUDA_VERSION=12.1.0
+PYTORCH_VERSION=2.2.2
+ENV_NAME='audienceview'
+
+"$HOME/miniconda3/bin/conda" create -y -n "$ENV_NAME" -c defaults \
+    python="$PYTHON_VERSION" ipykernel ipywidgets widgetsnbextension ipython
+
+"$HOME/miniconda3/bin/mamba" install -y -n "$ENV_NAME" -c "nvidia/label/cuda-$CUDA_VERSION" \
+    cuda="$CUDA_VERSION"
+
+"$HOME/miniconda3/bin/mamba" install -y -n "$ENV_NAME" -c pytorch -c nvidia -c defaults \
+    pytorch="$PYTORCH_VERSION" torchtext torchdata torchtriton pytorch-cuda
+
+"$HOME/miniconda3/bin/mamba" install -y -n "$ENV_NAME" -c conda-forge \
+    altair google-api-python-client hdbscan matplotlib nltk numpy openai \
+    pandas scipy sqlalchemy statsmodels streamlit tqdm transformers \
+    umap-learn wordcloud