Skip to content

GitLab

Commit 2628cedb authored by Rick Ho's avatar Rick Ho
Browse files

fix pytorch docker link in readme

parent dd7cbfc4
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
README.md
View file @ 2628cedb
...@@ -31,7 +31,8 @@ it, pass environment variable `USE_NCCL=1` to the setup script. ...@@ -31,7 +31,8 @@ it, pass environment variable `USE_NCCL=1` to the setup script.
Note that an extra NCCL developer package is needed, which has to be consistant Note that an extra NCCL developer package is needed, which has to be consistant
with your PyTorch's NCCL version, which can be inspected by running with your PyTorch's NCCL version, which can be inspected by running
`torch.cuda.nccl.version()`. The [official PyTorch docker image]() is `torch.cuda.nccl.version()`. The
[official PyTorch docker image](https://hub.docker.com/r/pytorch/pytorch) is
recommended, as the environment is well-setup there. Otherwise, you can access recommended, as the environment is well-setup there. Otherwise, you can access
the [download link of all NCCL the [download link of all NCCL
versions](https://developer.nvidia.com/nccl/nccl-legacy-downloads) to download versions](https://developer.nvidia.com/nccl/nccl-legacy-downloads) to download
......
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