Skip to content

Commit

Permalink
Merge branch 'sven-dev' into 'main'
Browse files Browse the repository at this point in the history
Amended docker/Dockerfile and edited notebooks/README.md

See merge request nvidia/trtorch/Torch-TensorRT-Preview!6
  • Loading branch information
narendasan committed Oct 23, 2021
2 parents 53ac60c + 2084b02 commit 78636eb
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
2 changes: 2 additions & 0 deletions docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -47,5 +47,7 @@ ENV PATH /opt/conda/lib/python3.8/site-packages/torch_tensorrt/bin:${PATH}

WORKDIR /workspace
RUN mv /workspace/torch_tensorrt /opt/pytorch/torch_tensorrt
RUN mkdir torch_tensorrt
RUN ln -s /opt/pytorch/torch_tensorrt/notebooks /workspace/torch_tensorrt/notebooks

CMD /bin/bash
3 changes: 2 additions & 1 deletion notebooks/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,8 @@ in, for example:
```http://[host machine]:8888/?token=aae96ae9387cd28151868fee318c3b3581a2d794f3b25c6b```


Within the container, the notebooks themselves are located at `/workspace/Torch-TensorRT/notebooks`.
Within the container, the notebooks themselves are located at `/workspace/torch_tensorrt/notebooks`. To reach them in Jupyter, click on the folder marked
`torch_tensorrt`, then the folder marked `notebooks`.

## 2. Notebook list

Expand Down

0 comments on commit 78636eb

Please sign in to comment.