diff --git a/src/ts2k/requirements.txt b/src/ts2k/requirements.txt
index f36135917..0828c1d91 100644
--- a/src/ts2k/requirements.txt
+++ b/src/ts2k/requirements.txt
@@ -1,4 +1,4 @@
 wheel==0.36.2
 sexpdata==0.0.3
-torch==1.8.0+cu111
+torch==1.8.1+cu111
 editdistance
diff --git a/test/builds/test_pytest.cmd b/test/builds/test_pytest.cmd
index 34974dd39..4e96d52ee 100644
--- a/test/builds/test_pytest.cmd
+++ b/test/builds/test_pytest.cmd
@@ -7,7 +7,7 @@ REM New resolver with PyTorch version notation https://pip.pypa.io/en/latest/use
 REM Too big, throw memory exception https://stackoverflow.com/a/31526029/35544
 echo Installing dependencies...
 python -m pip install -r src/python/requirements.txt
-python -m pip install --use-deprecated=legacy-resolver --no-cache-dir pytest numpy torch==1.8.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html
+python -m pip install --use-deprecated=legacy-resolver --no-cache-dir pytest numpy torch==1.8.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html
 
 echo Installing ksc...
 cd ./src/python
@@ -22,7 +22,7 @@ cd ./src/ts2k
 python -m pip install --editable .
 cd ../..
 
-REM currently have backend issues on PyTorch 1.8.0 https://github.com/microsoft/knossos-ksc/issues/659
+REM currently memory issues in tests PyTorch 1.8.1 https://github.com/microsoft/knossos-ksc/issues/679
 
 REM echo Running pytest on ts2k
 REM python -m pytest test/ts2k
diff --git a/test/builds/test_pytest.sh b/test/builds/test_pytest.sh
index 141128d9a..04f1201e7 100644
--- a/test/builds/test_pytest.sh
+++ b/test/builds/test_pytest.sh
@@ -2,7 +2,7 @@ set -e
 
 echo Installing dependencies...
 python3 -m pip install -r src/python/requirements.txt
-python3 -m pip install pytest numpy torch==1.8.0+cu111 jax==0.1.57 jaxlib==0.1.37 -f https://download.pytorch.org/whl/torch_stable.html
+python3 -m pip install pytest numpy torch==1.8.1+cu111 jax==0.1.57 jaxlib==0.1.37 -f https://download.pytorch.org/whl/torch_stable.html
 
 echo Installing ksc...
 cd ./src/python