Skip to content

Commit d9e0f37

Browse files
authored
Update user_install.dockerfile
1 parent 636afaa commit d9e0f37

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

.devcontainer/docker_include/user_install.dockerfile

+6-1
Original file line numberDiff line numberDiff 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+
3540
USER $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

0 commit comments

Comments
 (0)