From d960552677d23112a1bd053e5ea360469686bb7b Mon Sep 17 00:00:00 2001 From: boba_and_beer Date: Wed, 27 Jan 2021 01:11:41 +1100 Subject: [PATCH 1/2] Update README.md --- README.md | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index d7eaf7a..49db34c 100644 --- a/README.md +++ b/README.md @@ -17,11 +17,12 @@ CLIP (Contrastive Language-Image Pre-Training) is a neural network trained on a First, [install PyTorch 1.7.1](https://pytorch.org/get-started/locally/) and torchvision, as well as small additional dependencies. On a CUDA GPU machine, the following will do the trick: ```bash -$ conda install --yes -c pytorch pytorch=1.7.1 torchvision cudatoolkit=11.0 -$ pip install ftfy regex tqdm +$ pip install clip-by-openai +``` +Or if you want cuda installation: +```bash +$ pip install clip-by-openai[cuda] ``` - -Replace `cudatoolkit=11.0` above with the appropriate CUDA version on your machine or `cpuonly` when installing on a machine without a GPU. ```python import torch From 257e92e291f22a2240a9e04f1cd96e88147b7300 Mon Sep 17 00:00:00 2001 From: boba_and_beer Date: Wed, 27 Jan 2021 01:12:22 +1100 Subject: [PATCH 2/2] Update README.md --- README.md | 2 -- 1 file changed, 2 deletions(-) diff --git a/README.md b/README.md index 49db34c..9bc72d2 100644 --- a/README.md +++ b/README.md @@ -14,8 +14,6 @@ CLIP (Contrastive Language-Image Pre-Training) is a neural network trained on a ## Usage -First, [install PyTorch 1.7.1](https://pytorch.org/get-started/locally/) and torchvision, as well as small additional dependencies. On a CUDA GPU machine, the following will do the trick: - ```bash $ pip install clip-by-openai ```