AI_ENGINE_0.5/README.md

MG_ENGINE_SIM

위험성 평가기반 작업자 안전관리 및 감독 협조 기술 개발 (시뮬레이터)

System Requirements

- ubuntu 20.04
- python 3.8
- python packages : requirements.txt

Pytorch, Torchvision install

• conda(CUDA 11.7 기준)

  conda install pytorch torchvision pytorch-cuda=11.7 -c pytorch -c nvidia
  

• pip

  pip3 install torch torchvision torchaudio
  

• ARM architecture

  JetPack 5.0.2(L4T R35.1.0) 기준

  1. pytorch(version 1.12)

     wget https://developer.download.nvidia.com/compute/redist/jp/v50/pytorch/torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl
     apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev 
     pip3 install Cython 
     pip3 install torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl
     

  2. torchvision(version 0.13)

      sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
      git clone --branch release/0.13 https://github.com/pytorch/vision torchvision 
      cd torchvision
      export BUILD_VERSION=0.13.0
      python3 setup.py install --user