diff --git a/docker/Dockerfile.ubuntu_base b/docker/Dockerfile.ubuntu_base index 903c9930..4ce89be0 100644 --- a/docker/Dockerfile.ubuntu_base +++ b/docker/Dockerfile.ubuntu_base @@ -204,11 +204,11 @@ 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