frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Note that the Docker Hub image is not up-to-date. (#6434)

Browse Source 
Fixes #6397.
This commit is contained in:
Richard Zou
2018-04-16 14:31:34 -04:00
committed by Edward Z. Yang
parent 7ef14bf04c
commit 4d0097fab8
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
README.md
View File

@ -226,7 +226,8 @@ Dockerfile is supplied to build images with cuda support and cudnn v7. Build as
docker build -t pytorch -f docker/pytorch/Dockerfile .
```
Alternatively, if you want to use a runtime image, you can use the pre-built one from Docker Hub and run with nvidia-docker:
You can also pull a pre-built docker image from Docker Hub and run with nvidia-docker,
but this is not currently maintained and will pull PyTorch 0.2.
```
nvidia-docker run --rm -ti --ipc=host pytorch/pytorch:latest
```