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