Does conda install of deepmd support Platform: linux-aarch64?

[robinzyb]

Summary

no

DeePMD-kit Version

x

Backend and its version

x

Python Version, CUDA Version, GCC Version, LAMMPS Version, etc

No response

Details

I tried to install deepmd-kit using conda
conda create -n deepmd deepmd-kit lammps horovod -c conda-forge
And the message is

Channels:
 - conda-forge
 - defaults
Platform: linux-aarch64
Collecting package metadata (repodata.json): done
Solving environment: failed

PackagesNotFoundError: The following packages are not available from current channels:

  - lammps
  - horovod
  - deepmd-kit

Current channels:

  - https://conda.anaconda.org/conda-forge
  - defaults

To search for alternate channels that may provide the conda package you're
looking for, navigate to

    https://anaconda.org

and use the search bar at the top of the page.

I found there is only linux-64 not linux-aarch64. I am not sure if the error is caused by the platform. if so, could we add support on linux-aarch64

[njzjz]

xref: conda-forge/tensorflow-feedstock#136

[robinzyb]

thanks. This is required by newest hpc in CSCS (https://www.cscs.ch/computers/decommissioned/toedi). Looking forward to the update.
Our new project will start at this new hpc in October. The worst case would be compiling dpmd from scratch.

[robinzyb]

Hi, aarch64 is also required by the docker image. Does the docker iamge also suffer from the problem of upstream (tensorflow)? I only found x86 docker image for tensorflow in dockerhub (https://hub.docker.com/r/tensorflow/tensorflow/tags?page=2&page_size=&ordering=&name=)

[njzjz]

Hi, aarch64 is also required by the docker image. Does the docker iamge also suffer from the problem of upstream (tensorflow)? I only found x86 docker image for tensorflow in dockerhub (https://hub.docker.com/r/tensorflow/tensorflow/tags?page=2&page_size=&ordering=&name=)

Currently, we build docker images just using the offline installer, maintained at https://github.com/deepmd-kit-recipes/docker-images.

[robinzyb]

Hello, I checked the dockerfile. The offline installer only has linux x86_64.sh. How do you build the offline installer? Do we need to build an arm64 offline installer?

[njzjz]

How do you build the offline installer?

The offline package is built at https://github.com/deepmd-kit-recipes/installer, which is a collection of conda packages.

[robinzyb]

Thank you. It seems everything goes back to the conda install deepmd-kit where we need the aarch64 for tensorflow. should fix the upstream problem first.

[robinzyb]

memo: the containerfile for linux-aarch64 has been made by the cscs engineer, and the software performs well.