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 .
21
21
Then launch the container with:
22
22
23
23
```
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
25
25
```
26
26
27
27
Within the docker interactive bash session, start Jupyter with
28
28
29
29
```
30
- cd /opt/pytorch/torch_tensorrt /notebooks
30
+ cd /Torch-TensorRT /notebooks
31
31
jupyter notebook --allow-root --ip 0.0.0.0 --port 8888
32
32
```
33
33
@@ -40,7 +40,7 @@ in, for example:
40
40
``` http://[host machine]:8888/?token=aae96ae9387cd28151868fee318c3b3581a2d794f3b25c6b ```
41
41
42
42
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 ` .
44
44
45
45
## 2. Notebook list
46
46
You can’t perform that action at this time.
0 commit comments