Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 04:01:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

support fairseq for py311 in ubuntu_base docker

Browse Source
This commit is contained in:
xingjun.wxj
2025-03-17 12:56:52 +08:00
parent a5acaa01d0
commit a7901b3168
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
docker/Dockerfile.ubuntu_base
View File

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