pytorchcuda镜像
时间: 2023-08-08 11:07:31 浏览: 198
您好!对于使用PyTorch在CUDA上进行训练和推理的需求,您可以使用以下镜像:
```
docker pull pytorch/pytorch:latest
```
这是PyTorch官方维护的Docker镜像,其中包含了PyTorch在CUDA上的支持。您可以根据需要选择不同的版本,比如`pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime`等。
希望这能帮到您!如果还有其他问题,请随时提问。
阅读全文