Skip to content

Commit dbeb7e3

Browse files
committed
Merge branch 'main' into 'main'
fix notebook for API change See merge request nvidia/trtorch/Torch-TensorRT-Preview!9
2 parents 2d2656c + 7b4e504 commit dbeb7e3

File tree

5 files changed

+769
-743
lines changed

5 files changed

+769
-743
lines changed

notebooks/README.md

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,18 +14,20 @@ git clone https://github.com/NVIDIA/Torch-TensorRT
1414
Next, build the NVIDIA Torch-TensorRT container (from repo root):
1515

1616
```
17+
cd Torch-TensorRT
1718
docker build -t torch_tensorrt -f ./docker/Dockerfile .
1819
```
1920

2021
Then launch the container with:
2122

2223
```
23-
docker run --runtime=nvidia -it --rm --ipc=host --net=host torch_tensorrt
24+
docker run --gpus=all --rm -it -v $PWD:/Torch-TensorRT --net=host torch_tensorrt bash
2425
```
2526

2627
Within the docker interactive bash session, start Jupyter with
2728

2829
```
30+
cd /Torch-TensorRT/notebooks
2931
jupyter notebook --allow-root --ip 0.0.0.0 --port 8888
3032
```
3133

@@ -38,8 +40,7 @@ in, for example:
3840
```http://[host machine]:8888/?token=aae96ae9387cd28151868fee318c3b3581a2d794f3b25c6b```
3941

4042

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

4445
## 2. Notebook list
4546

0 commit comments

Comments
 (0)