We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see ourdocumentation.
There was an error while loading.Please reload this page.
1 parent78d8b08 commit6ed3a51Copy full SHA for 6ed3a51
.github/container/Dockerfile.maxtext
@@ -23,6 +23,11 @@ git-clone.sh ${URLREF_MAXTEXT} ${SRC_PATH_MAXTEXT}
23
git-clone.sh ${URLREF_JETSTREAM} ${SRC_PATH_JETSTREAM}
24
EOF
25
26
+# WAR: tunix package is not needed for Nvidia GPU support
27
+RUN <<"EOF" bash -ex -o pipefail
28
+sed -i '/tunix/d' ${SRC_PATH_MAXTEXT}/src/install_maxtext_extra_deps/extra_deps_from_github.txt
29
+EOF
30
+
31
RUN <<"EOF" bash -ex -o pipefail
32
echo "-e file://${SRC_PATH_MAXTEXT}" >> /opt/pip-tools.d/requirements-maxtext.in
33
echo "-r ${SRC_PATH_MAXTEXT}/src/install_maxtext_extra_deps/extra_deps_from_github.txt" >> /opt/pip-tools.d/requirements-maxtext.in