add torch sciplot env

This commit is contained in:
2024-05-26 14:23:25 +08:00
parent 2c58a44cde
commit 98265022aa

View File

@@ -174,5 +174,14 @@ cargo install evcxr_jupyter
evcxr_jupyter --install evcxr_jupyter --install
EOT 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}" WORKDIR "${HOME}"