diff --git a/requirements.txt b/requirements.txt index 52126ca7af..d3ece66ff8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,3 +5,4 @@ scikit-learn==0.20.2 tqdm==4.32.1 twine==1.13.0 numpy==1.16.4 +torch>=1.1.0 \ No newline at end of file