OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

fixed README to reflect docker image name (#1751)

Browse Source
This commit is contained in:
Matt 2017-06-07 15:49:40 -04:00 committed by Soumith Chintala
parent e57eef4bcb
commit a9bd1de9e9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -207,7 +207,7 @@ docker build -t pytorch . -f tools/docker/Dockerfile_runtime
``` ```
and run with nvidia-docker: and run with nvidia-docker:
``` ```
nvidia-docker run --rm -ti --ipc=host pytorch-cudnnv6 nvidia-docker run --rm -ti --ipc=host pytorch
``` ```
Please note that PyTorch uses shared memory to share data between processes, so if torch multiprocessing is used (e.g. Please note that PyTorch uses shared memory to share data between processes, so if torch multiprocessing is used (e.g.
for multithreaded data loaders) the default shared memory segment size that container runs with is not enough, and you for multithreaded data loaders) the default shared memory segment size that container runs with is not enough, and you