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

Update readme with docker cmd (#2501)

Browse Source 
* update docker command in readme to use pre-built images

* correct spelling of Docker Hub

* Update README.md
This commit is contained in:
Alykhan Tejani 2017-08-21 13:52:26 +01:00 committed by Soumith Chintala
parent d2b8d3f8f7
commit de24bb4b66
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
README.md
View File

@ -199,15 +199,9 @@ Dockerfile is supplied to build images with cuda support and cudnn v6. Build as
```
docker build -t pytorch .
```
Alternatively, if you want a runtime image, build with
Alternatively, if you want to use a runtime image, you can use the pre-built one from Docker Hub and run with nvidia-docker:
```
docker build -t pytorch . -f tools/docker/Dockerfile_runtime
```
and run with nvidia-docker:
```
nvidia-docker run --rm -ti --ipc=host pytorch
nvidia-docker run --rm -ti --ipc=host pytorch/pytorch:latest
```
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