diff --git a/Makefile b/Makefile
index 356d48b..766a4f2 100644
--- a/Makefile
+++ b/Makefile
@@ -16,7 +16,7 @@ black: ## [Local development] Auto-format python code using black
 build-pex:
 	python3 -m venv .pexing
 	. .pexing/bin/activate && python -m pip install -U pip && python -m pip install pex
-	. .pexing/bin/activate && python -m pex --layout packed -f https://download.pytorch.org/whl/cu113/torch_stable.html setuptools gcsfs charset-normalizer s3fs pyspark torch==1.10.2+cu113 torchvision==0.11.3+cu113 . -o clip_retrieval.pex -v
+	. .pexing/bin/activate && python -m pex --layout packed setuptools gcsfs charset-normalizer s3fs pyspark torch torchvision . -o clip_retrieval.pex -v
 	rm -rf .pexing
 	tar czf clip_retrieval_torch.tgz clip_retrieval.pex/.deps/torch-*
 	tar czf clip_retrieval.tgz --exclude clip_retrieval.pex/.deps/torch-* clip_retrieval.pex
diff --git a/requirements.txt b/requirements.txt
index 20fe732..e6357d2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ img2dataset>=1.25.5,<2
 clip-anytorch>=2.5.0,<3
 tqdm>=4.62.3,<5
 fire>=0.4.0,<0.6.0
-torch>=1.7.1,<2
+torch>=1.7.1,<3
 torchvision>=0.10.1,<2
 numpy>=1.19.5,<2
 faiss-cpu>=1.7.2,<2