Skip to content

Commit

Permalink
Fix command highlight in README (#3521)
Browse files Browse the repository at this point in the history
  • Loading branch information
todpole3 authored and soumith committed Nov 7, 2017
1 parent 6dd87dc commit d2ddbaa
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -202,9 +202,11 @@ MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py install
Dockerfile is supplied to build images with cuda support and cudnn v6. Build as usual
```
docker build -t pytorch .
```

Dockerfile to build with cuda 9 and cudnn v7 (with Volta support) is in tools/docker, the build command is

```
docker build -t pytorch_cuda9 -f tools/docker/Dockerfile9 .
```
Alternatively, if you want to use a runtime image, you can use the pre-built one from Docker Hub and run with nvidia-docker:
Expand Down

0 comments on commit d2ddbaa

Please sign in to comment.