"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "06fd4277973d9ca14f7cdf364943c28de95e5510"
Unverified Commit 606c3dcc authored by mcarilli's avatar mcarilli Committed by GitHub
Browse files

Update README.md

parent 5b29cc13
...@@ -12,7 +12,7 @@ Alternatively, you can supply your own base image via the `BASE_IMAGE` build-arg ...@@ -12,7 +12,7 @@ Alternatively, you can supply your own base image via the `BASE_IMAGE` build-arg
`-devel` image for Pytorch 1.0 and later from the `-devel` image for Pytorch 1.0 and later from the
[official Pytorch Dockerhub](https://hub.docker.com/r/pytorch/pytorch) may be used: [official Pytorch Dockerhub](https://hub.docker.com/r/pytorch/pytorch) may be used:
``` ```
docker build --build-arg BASE_IMAGE=pytorch/pytorch:nightly-devel-cuda10.0-cudnn7 -t new_image_with_apex . docker build --build-arg BASE_IMAGE=1.3-cuda10.1-cudnn7-devel -t new_image_with_apex .
``` ```
If you want to rebuild your image, and force the latest Apex to be cloned and installed, make any small change to the `SHA` variable in **Dockerfile**. If you want to rebuild your image, and force the latest Apex to be cloned and installed, make any small change to the `SHA` variable in **Dockerfile**.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment