File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed
.devcontainer/docker_include Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -32,6 +32,11 @@ RUN pip install --no-cache-dir \
3232# COPY requirements.txt ${COLCON_WS}/src/requirements.txt
3333# RUN pip install -r ${COLCON_WS}/src/requirements.txt && rm ${COLCON_WS}/src/requirements.txt
3434
35+
36+ RUN pip install nvfuser-cu118-torch24
37+ RUN export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.10/dist-packages/torch/lib:$LD_LIBRARY_PATH
38+
39+
3540USER $USER
3641
3742# Copy the script to checkout public git repos and make it executable
@@ -41,4 +46,4 @@ RUN /bin/bash -c '${COLCON_WS}/src/install_external_ros_packages.sh ${COLCON_WS}
4146 sudo rm -f ${COLCON_WS}/src/install_external_ros_packages.sh && \
4247 sudo rm -f -r ${COLCON_WS}/src/repos
4348
44- USER root
49+ USER root
You can’t perform that action at this time.
0 commit comments