localhorst
/
CLIP_on_Tesla_K20Xm
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge branch 'feature/make_installable' of github.com:boba-and-beer/CLIP into feature/make_installable

This commit is contained in:
Jacky 2021-01-27 01:26:42 +11:00
parent d1b57d8b97 257e92e291
commit 3e228f401f
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
README.md
View File

@ -14,14 +14,13 @@ 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
$ 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