mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
Update Dockerfile build instructions for new layout. (#6051)
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
This commit is contained in:
parent
5583c12888
commit
b5fa9a82c8
|
|
@ -222,7 +222,7 @@ python setup.py install
|
|||
|
||||
Dockerfile is supplied to build images with cuda support and cudnn v7. Build as usual
|
||||
```
|
||||
docker build -t pytorch .
|
||||
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:
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user