File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -21,13 +21,13 @@ docker build -t torch_tensorrt -f ./docker/Dockerfile .
2121Then 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
2727Within the docker interactive bash session, start Jupyter with
2828
2929```
30- cd /opt/pytorch/torch_tensorrt /notebooks
30+ cd /Torch-TensorRT /notebooks
3131jupyter 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
You can’t perform that action at this time.
0 commit comments