We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bd9c29a commit 44e4ffaCopy full SHA for 44e4ffa
docker/Dockerfile
@@ -94,6 +94,11 @@ RUN bash ./docker/dist-build.sh
94
FROM base as torch-tensorrt
95
96
COPY . /opt/torch_tensorrt
97
+
98
+# Symlink the path pyenv is using for python with the /opt directory for package sourcing
99
+RUN mkdir -p "/opt/python3/" &&\
100
+ ln -s "`pyenv which python | xargs dirname | xargs dirname`/lib/python$PYTHON_VERSION/site-packages" "/opt/python3/"
101
102
COPY --from=torch-tensorrt-builder /workspace/torch_tensorrt/src/py/dist/ .
103
104
RUN cp /opt/torch_tensorrt/docker/WORKSPACE.docker /opt/torch_tensorrt/WORKSPACE
0 commit comments