Docker Hub 地址
已经编译好的 Docker Image 在 Docker Hub 上,地址是 https://hub.docker.com/r/anibali/pytorch/。例如,你想使用 CUDA 9.2 版本,则使用下列命令:
1 |
$ docker pull anibali/pytorch:cuda-9.2 |
现在支持的版本有:
Image tag | CUDA | PyTorch |
---|---|---|
no-cuda |
None | 0.4.1 |
cuda-9.2 |
9.2 | 0.4.1 |
cuda-9.0 |
9.0 | 0.4.1 |
cuda-8.0 |
8.0 | 0.4.1 |
下面的版本也有,但不再提供支持:
Image tag | CUDA | PyTorch |
---|---|---|
cuda-9.1 |
9.1 | 0.4.0 |
cuda-7.5 |
7.5 | 0.3.0 |
如何运行 PyTorch?
具体运行方法请查看