Skip to content

Commit

Permalink
Double quote PyTorch dependency
Browse files Browse the repository at this point in the history
  • Loading branch information
@quantumsheep
quantumsheep committed Oct 13, 2023
1 parent 125e6d9 commit 15e535b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deploy/Dockerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ RUN apt update && \
ENV TORCH_CUDA_ARCH_LIST "7.0;7.2;7.5;8.0;8.6;8.9;9.0"

# We have to manually install Torch otherwise apex & xformers won't build
RUN pip3 install torch==2.2.0.dev20231013+cu118 --index-url https://download.pytorch.org/whl/nightly/cu118
RUN pip3 install "torch==2.2.0.dev20231013+cu118" --index-url https://download.pytorch.org/whl/nightly/cu118

# This build is slow but NVIDIA does not provide binaries. Increase MAX_JOBS as needed.
RUN git clone https://github.com/NVIDIA/apex && \
Expand Down

0 comments on commit 15e535b

Please sign in to comment.