Skip to content

Commit 8511c73

Browse files
committed
Reformatted the notebook README file
Added headers and regrouped the notebook README file for easier navigation, per @narendasan's suggestion Signed-off-by: Sven Chilton <[email protected]>
1 parent 9cf7b34 commit 8511c73

File tree

1 file changed

+13
-7
lines changed

1 file changed

+13
-7
lines changed

notebooks/README.md

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@ Next, navigate to the repo's root directory:
1717
cd Torch-TensorRT
1818
```
1919

20+
### a. Using the NGC PyTorch container
21+
2022
At this point, we recommend pulling the [PyTorch container](https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch)
2123
from [NVIDIA GPU Cloud](https://catalog.ngc.nvidia.com/) as follows:
2224

@@ -29,12 +31,6 @@ where ```yy``` indicates the last two numbers of a calendar year, and
2931
```mm``` indicates the month in two-digit numerical form, if you wish
3032
to pull a different version of the container.
3133

32-
Alternatively, to build the container from source, run
33-
34-
```
35-
docker build -t torch_tensorrt -f ./docker/Dockerfile .
36-
```
37-
3834
The NGC PyTorch container ships with the Torch-TensorRT tutorial notebooks.
3935
Therefore, you can run the container and the notebooks therein without
4036
mounting the repo to the container. To do so, run
@@ -50,12 +46,22 @@ If, however, you wish for your work in the notebooks to persist, use the
5046
docker run --gpus=all --rm -it -v $PWD:/Torch-TensorRT --net=host --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 nvcr.io/nvidia/pytorch:21.12-py3 bash
5147
```
5248

53-
If you're using a container built from source, run this instead:
49+
### b. Building a Torch-TensorRT container from source
50+
51+
Alternatively, to build the container from source, run
52+
53+
```
54+
docker build -t torch_tensorrt -f ./docker/Dockerfile .
55+
```
56+
57+
To run this container, enter the following command:
5458

5559
```
5660
docker run --gpus=all --rm -it -v $PWD:/Torch-TensorRT --net=host --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 torch_tensorrt:latest bash
5761
```
5862

63+
### c. Running the notebooks inside the container
64+
5965
Within the docker interactive bash session, proceed to the notebooks.
6066
To use the notebooks which ship with the container, run
6167

0 commit comments

Comments
 (0)