Fixed tensorflow-io for noavx

snowzach · Dec 30, 2021 · eb4d0bd · eb4d0bd
1 parent 3eac71e
commit eb4d0bd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/Dockerfile.amd64-noavx b/Dockerfile.amd64-noavx
@@ -19,8 +19,11 @@ RUN python3 -m pip --no-cache-dir install --upgrade \
     pip \
     setuptools
 
-# Install Tensorflow - NOAVX
-RUN python3 -m pip install --no-cache-dir https://github.com/snowzach/tensorflow-multiarch/releases/download/v2.7.0/tensorflow-2.7.0+noavx-cp38-cp38-linux_x86_64.whl
+# Install Tensorflow NOAVX and Deps
+RUN python3 -m pip install --no-cache-dir \
+    https://github.com/snowzach/tensorflow-multiarch/releases/download/v2.7.0/tensorflow_io_gcs_filesystem-0.23.1+noavx-cp38-cp38-linux_x86_64.whl \
+    https://github.com/snowzach/tensorflow-multiarch/releases/download/v2.7.0/tensorflow_io-0.23.1+noavx-cp38-cp38-linux_x86_64.whl \
+    https://github.com/snowzach/tensorflow-multiarch/releases/download/v2.7.0/tensorflow-2.7.0+noavx-cp38-cp38-linux_x86_64.whl
 
 ADD https://raw.githubusercontent.com/tensorflow/tensorflow/master/tensorflow/tools/dockerfiles/bashrc /etc/bash.bashrc
 RUN chmod a+rwx /etc/bash.bashrc