Relax requirements

Signed-off-by: Beat Buesser <[email protected]>

Author: Beat Buesser

requirements.txt

@@ -21,7 +21,7 @@ keras>=2.2.5
 tensorflow-addons~=0.11.0
 mxnet==1.6.0
 --find-links https://download.pytorch.org/whl/torch_stable.html
-torch~=1.8.0
+torch==1.8.0+cpu
 torchaudio~=0.8.0
 torchvision~=0.9.0
 catboost==0.24.1