Skip to content

Commit 7761edb

Browse files
vinhngxnarendasan
authored andcommitted
fix readme
Signed-off-by: Vinh Nguyen <[email protected]>
1 parent 8fa433a commit 7761edb

File tree

1 file changed

+37
-0
lines changed

1 file changed

+37
-0
lines changed

notebooks/README.md

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
# Jupyter demo notebooks
2+
This folder contains demo notebooks for the TRTorch.
3+
4+
## 1. Requirements
5+
6+
The most convenient way to run these notebooks is via a docker container, which provides a self-contained, isolated and re-producible environment for all experiments.
7+
8+
First, clone the repository:
9+
10+
```
11+
git clone https://github.com/NVIDIA/TRTorch
12+
```
13+
14+
Next, build the NVIDIA TRTorch container:
15+
16+
```
17+
docker build -t trtorch -f Dockerfile.notebook .
18+
```
19+
20+
Then launch the container with:
21+
22+
```
23+
docker run --runtime=nvidia -it --rm --ipc=host --net=host trtorch
24+
```
25+
where `/path/to/dataset` is the path on the host machine where the data was/is to be downloaded. More on data set preparation in the next section. `/path/to/results` is wher the trained model will be stored.
26+
27+
Within the docker interactive bash session, start Jupyter with
28+
29+
```
30+
jupyter notebook --allow-root --ip 0.0.0.0 --port 8888
31+
```
32+
33+
Then open the Jupyter GUI interface on your host machine at http://localhost:8888. Within the container, this notebook itself is located at `/workspace/TRTorch/notebooks`.
34+
35+
## 2. Notebook list
36+
37+
- [LeNet-example.ipynb](LeNet-example.ipynb): simple example on a LeNet network.

0 commit comments

Comments
 (0)