From 98265022aa14f2e7b94aa541698ed61169116522 Mon Sep 17 00:00:00 2001
From: hotwa <pylyzeng@gmail.com>
Date: Sun, 26 May 2024 14:23:25 +0800
Subject: [PATCH] add torch sciplot env

---
 spawnerdockerfile/Dockerfile.base-notebook | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/spawnerdockerfile/Dockerfile.base-notebook b/spawnerdockerfile/Dockerfile.base-notebook
index cf74c55..bd39340 100755
--- a/spawnerdockerfile/Dockerfile.base-notebook
+++ b/spawnerdockerfile/Dockerfile.base-notebook
@@ -174,5 +174,14 @@ cargo install evcxr_jupyter
 evcxr_jupyter --install
 EOT
 
+RUN <<EOT
+#!/bin/bash
+sudo apt update
+sudo apt install -y texlive-full
+mamba create -n torch pytorch torchvision torchaudio python=3 ipython nglview requests scienceplots autopep8 ipykernel\
+ pandas numpy matplotlib scipy seaborn orange3 -c pytorch -c nvidia -c conda-forge -y 
+mamba run -n torch python -m ipykernel install --user --name="torch" --display-name="torch_env"
+EOT
+
 WORKDIR "${HOME}"