install issue fix

conda to install cpuonly instead of environment script
3 years ago · 572e2c3de8
parent b8df6e00dc
commit 572e2c3de8
2 changed files with 1 additions and 5 deletions
--- a/README.md
+++ b/README.md
@ -25,6 +25,7 @@ A suitable [conda](https://conda.io/) environment named `ldm` can be created
 and activated with:

 ```
+conda install pytorch torchvision torchaudio cpuonly -c pytorch
 conda env create -f environment-cpuonly.yaml
 conda activate ldm
 ```
--- a/environment-cpuonly.yaml
+++ b/environment-cpuonly.yaml
@ -6,11 +6,6 @@ dependencies:
  - python=3.8.5
  - pip=20.3
  - numpy=1.19.2
-  - pytorch
-  - torchvision
-  - torchaudio
-  - cpuonly
-  - -c pytorch
  - pip:
    - albumentations==0.4.3
    - diffusers