Skip to content

Commit 7b4e504

Browse files
committed
fix notebook location: mount directory to docker, so changes to notebooks are saved
1 parent 7bb7850 commit 7b4e504

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

notebooks/README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,13 +21,13 @@ docker build -t torch_tensorrt -f ./docker/Dockerfile .
2121
Then launch the container with:
2222

2323
```
24-
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
2525
```
2626

2727
Within the docker interactive bash session, start Jupyter with
2828

2929
```
30-
cd /opt/pytorch/torch_tensorrt/notebooks
30+
cd /Torch-TensorRT/notebooks
3131
jupyter notebook --allow-root --ip 0.0.0.0 --port 8888
3232
```
3333

@@ -40,7 +40,7 @@ in, for example:
4040
```http://[host machine]:8888/?token=aae96ae9387cd28151868fee318c3b3581a2d794f3b25c6b```
4141

4242

43-
Within the container, the notebooks themselves are located at `/opt/pytorch/torch_tensorrt/notebooks`.
43+
Within the container, the notebooks themselves are located at `/Torch-TensorRT/notebooks`.
4444

4545
## 2. Notebook list
4646

0 commit comments

Comments
 (0)