Skip to content

H
Hydra Node Http
Commit 6e86dbfb authored by novelailab's avatar novelailab
Browse files
Options

Enable FORCE_CUDA

parent e64da730
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
Dockerfile
View file @ 6e86dbfb
......@@ -33,6 +33,7 @@ RUN pip3 install https://www.dropbox.com/s/8ozhhbo1g7y5dsz/basedformer-f7c8c4fe1
#built DS
#RUN pip3 install https://www.dropbox.com/s/euzpgpfrs9isf1z/deepspeed-0.7.3%2B55b7b9e0-cp38-cp38-linux_x86_64.whl?dl=1
RUN pip3 install faiss-cpu sentence_transformers
ENV FORCE_CUDA 1
RUN pip3 install 'git+https://github.com/facebookresearch/xformers@51dd119#egg=xformers'
#RUN pip install -U --pre triton
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment