From b8df6e00dc7937343f3f9479bcaac703f640e894 Mon Sep 17 00:00:00 2001 From: Christopher Williams Date: Thu, 25 Aug 2022 15:50:29 -0400 Subject: [PATCH] added cpuonly environment script --- README.md | 14 +++++++------- environment-cpuonly.yaml | 33 +++++++++++++++++++++++++++++++++ 2 files changed, 40 insertions(+), 7 deletions(-) create mode 100644 environment-cpuonly.yaml diff --git a/README.md b/README.md index c9e6c3b..765d4aa 100644 --- a/README.md +++ b/README.md @@ -25,16 +25,16 @@ A suitable [conda](https://conda.io/) environment named `ldm` can be created and activated with: ``` -conda env create -f environment.yaml +conda env create -f environment-cpuonly.yaml conda activate ldm ``` -You can also update an existing [latent diffusion](https://github.com/CompVis/latent-diffusion) environment by running - -``` -conda install pytorch torchvision -c pytorch -pip install transformers==4.19.2 diffusers invisible-watermark -pip install -e . +//You can also update an existing [latent diffusion](https://github.com/CompVis/latent-diffusion) environment by running +// +//``` +//conda install pytorch torchvision -c pytorch +//pip install transformers==4.19.2 diffusers invisible-watermark +//pip install -e . ``` diff --git a/environment-cpuonly.yaml b/environment-cpuonly.yaml new file mode 100644 index 0000000..cb83a1b --- /dev/null +++ b/environment-cpuonly.yaml @@ -0,0 +1,33 @@ +name: ldm +channels: + - pytorch + - defaults +dependencies: + - python=3.8.5 + - pip=20.3 + - numpy=1.19.2 + - pytorch + - torchvision + - torchaudio + - cpuonly + - -c pytorch + - pip: + - albumentations==0.4.3 + - diffusers + - opencv-python==4.1.2.30 + - pudb==2019.2 + - invisible-watermark + - imageio==2.9.0 + - imageio-ffmpeg==0.4.2 + - pytorch-lightning==1.4.2 + - omegaconf==2.1.1 + - test-tube>=0.7.5 + - streamlit>=0.73.1 + - einops==0.3.0 + - torch-fidelity==0.3.0 + - transformers==4.19.2 + - torchmetrics==0.6.0 + - kornia==0.6 + - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers + - -e git+https://github.com/openai/CLIP.git@main#egg=clip + - -e .