软件要求如表2所示。
项目 |
版本 |
下载地址 |
|---|---|---|
Docker |
19.03.15 |
https://download.docker.com/linux/static/stable/aarch64/docker-19.03.15.tgz |
NVIDIA Container Toolkit |
Yum安装 |
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html |
PyTorch |
1.9 |
|
Cuda |
11.1 |
https://registry.hub.docker.com/r/nvidia/cuda-arm64/tags?page=1&ordering=last_updated&name=11.1 |
Cudnn |
8 |
https://developer.download.nvidia.cn/compute/machine-learning/repos/rhel8/sbsa/ |