From c0065a27ad170a67d9572f693292b594982ab9b2 Mon Sep 17 00:00:00 2001 From: Jong Wook Kim Date: Sun, 10 Apr 2022 13:39:41 -0700 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 5d2d20c..abc8a34 100644 --- a/README.md +++ b/README.md @@ -14,7 +14,7 @@ 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, and then install this repo as a Python package. On a CUDA GPU machine, the following will do the trick: +First, [install PyTorch 1.7.1](https://pytorch.org/get-started/locally/) (or later) and torchvision, as well as small additional dependencies, and then install this repo as a Python package. 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