49 lines
1.4 KiB
Markdown
49 lines
1.4 KiB
Markdown
# 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
|
|
```
|
|
|
|
### yolov5 install
|
|
DL/OD 폴더에 저장
|
|
```
|
|
cd DL
|
|
git clone https://github.com/ultralytics/yolov5.git OD
|
|
```
|
|
|
|
git commit id : 23c492321290266810e08fa5ee9a23fc9d6a571f
|
|
|
|
```
|
|
git reset hard -- 23c492321290266810e08fa5ee9a23fc9d6a571f
|
|
``` |