You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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
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