torchvision build manual for FUJITSU Software Compiler Package (v0.8.1) - fujitsu/vision GitHub Wiki

Requirements

  • FUJITSU Software Compiler Package is already installed.
  • The login node has access to the external network.
  • PyTorch is already installed. - Follow the PyTorch build manual instructions to install.
  • libjpeg is already installed. - It will be installed at the same time you install PyTorch according to the instructions in the PyTorch build manual.
  • Pillow is already installed. - It will be installed at the same time you install PyTorch according to the instructions in the PyTorch build manual.

Verified environment

Hardware Fujitsu FX1000 / FX700
OS RedHad 8.1 / Centos 8.1
Compiler FUJITSU Software Compiler Package
PyTorch v1.7.0
libjpeg v9d
Pillow v8.0.0

Preparation

  1. Checkout from Repository.
git clone https://github.com/fujitsu/vision.git
cd vision
git checkout -b fujitsu_v0.8.1_for_a64fx origin/fujitsu_v0.8.1_for_a64fx
  1. Environment Setting

Example.

export TCSDS_PATH=/opt/FJSVxtclanga/tcsds-1.2.27a

export CC="${TCSDS_PATH}/bin/fcc -Nclang -Kfast -Knolargepage -lpthread"
export CXX="${TCSDS_PATH}/bin/FCC -Nclang -Kfast -Knolargepage -lpthread"

export LD_LIBRARY_PATH=${TCSDS_PATH}/lib64:${LD_LIBRARY_PATH}
export PATH=${TCSDS_PATH}/bin:${PATH}

Build vision

Build binary files for A64FX.

For pytorch

python3 setup.py install

Copyright

Copyright RIKEN, Japan 2021 Copyright FUJITSU LIMITED 2021