remove tinycudann

This commit is contained in:
xingjun.wxj
2025-03-12 13:56:36 +08:00
parent 3bfa5f4ce5
commit b19471cccd

View File

@@ -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