diff --git a/requirements-cuda.txt b/requirements-cuda.txt
index 264c11a..cbb3279 100644
--- a/requirements-cuda.txt
+++ b/requirements-cuda.txt
@@ -3,7 +3,8 @@ h5py==2.9.0
 Keras==2.2.4
 opencv-python==4.0.0.21
 tensorflow-gpu==1.12.0
-plaidml-keras==0.6.0rc1
+plaidml==0.6.0rc1
+plaidml-keras==0.5.0
 scikit-image
 tqdm
 ffmpeg-python==0.1.17
diff --git a/requirements-opencl.txt b/requirements-opencl.txt
index d2a7db6..2a357c1 100644
--- a/requirements-opencl.txt
+++ b/requirements-opencl.txt
@@ -3,7 +3,8 @@ h5py==2.9.0
 Keras==2.2.4
 opencv-python==4.0.0.21
 tensorflow==1.12.0
-plaidml-keras==0.6.0rc1
+plaidml==0.6.0rc1
+plaidml-keras==0.5.0
 scikit-image
 tqdm
 ffmpeg-python==0.1.17