Fix Dockerfile.cuda

This commit is contained in:
henryruhs
2023-10-12 22:43:55 +02:00
parent c5d4c61e6f
commit 9f676af5a8
+1 -1
View File
@@ -15,4 +15,4 @@ RUN apt-get install ffmpeg -y
RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch .
RUN python install.py --torch cuda --onnxruntime cuda
RUN cd /usr/local/lib/python3.10/dist-packages/torch/lib && ln -s libnvrtc-672ee683 libnvrtc.so
RUN cd /usr/local/lib/python3.10/dist-packages/torch/lib && ln -s libnvrtc-672ee683.so.11.2 libnvrtc.so