From a7901b3168d99ada752bd62c7291f9d13e66afdf Mon Sep 17 00:00:00 2001 From: "xingjun.wxj" Date: Mon, 17 Mar 2025 12:56:52 +0800 Subject: [PATCH] support fairseq for py311 in ubuntu_base docker --- docker/Dockerfile.ubuntu_base | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/docker/Dockerfile.ubuntu_base b/docker/Dockerfile.ubuntu_base index 903c9930..fae19cfb 100644 --- a/docker/Dockerfile.ubuntu_base +++ b/docker/Dockerfile.ubuntu_base @@ -204,12 +204,13 @@ ARG TENSORFLOW_VERSION={tf_version} fi # This limits the cuda121 version - RUN if [ "$USE_GPU" = "True" ] ; then \ - pip install --no-cache-dir --force tinycudann==1.7 -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html; \ - else \ - echo 'cpu not install tinycudann'; \ - fi +# RUN if [ "$USE_GPU" = "True" ] ; then \ +# pip install --no-cache-dir --force tinycudann==1.7 -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html; \ +# else \ +# echo 'cpu not install tinycudann'; \ +# fi - RUN pip install --no-cache-dir fairseq + # RUN pip install --no-cache-dir fairseq # for py310 + RUN pip install --no-cache-dir https://github.com/liyaodev/fairseq/releases/download/v0.12.3.1/fairseq-0.12.3.1-cp311-cp311-linux_x86_64.whl # for py311 ENTRYPOINT []