본문 바로가기

컴퓨터

(66)
Super resolution 모델, HAT, inference 사용 정리 Ubuntu 20.04 ROCm AMD GPU Radeon VII 공식 github GitHub - XPixelGroup/HAT: Arxiv2022 - Activating More Pixels in Image Super-Resolution Transformer GitHub - XPixelGroup/HAT: Arxiv2022 - Activating More Pixels in Image Super-Resolution Transformer Arxiv2022 - Activating More Pixels in Image Super-Resolution Transformer - GitHub - XPixelGroup/HAT: Arxiv2022 - Activating More Pixels in Image Super-Re..
TensorRT Docker 사용 정리 Nvidia driver: 515.86 CUDA: 11.6 Docker: 20.10.21 1. Nvidia container toolkit 설치(Installation Guide — NVIDIA Cloud Native Technologies documentation) 1-1. package repository 추가 distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \ && curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \ && curl -s -L ht..
3080, Radeon vii, 6900xt, 딥러닝 (image classification) 학습 성능 비교 완벽하게 동일한 세팅(CPU/RAM/storage)에서 비교한 것이 아니므로 오로지 GPU의 성능을 비교했다고 하긴 어려움 적당한 비교(just curious)로 받아들이면 좋을듯 동일한 학습 모델, 데이터 셋에서의 학습 성능 비교 - 첫 2 epoch에 걸린 시간을 비교 Framework: Pytorch Model: efficientnet lite0 Batch size: 20 Image size: 384x384 데이터 수: 약 3만장 시스템: 결과: 결론 1. 역시 Nvidia가 좋다. 2. 학습속도는 조금 느릴지라도 AMD의 높은 GPU RAM은 무거운 모델 학습/구동에 좋을 수도 있겠다.
Ubuntu, ROCm, AMD GPU, Docker, Pytorch 환경에서 딥러닝 정리 ROCm이 설치된 Ubuntu 환경에서 Docker-Pytorch 활용 딥러닝에 대해 정리 Radeon VII 사용 Ubuntu 환경에 ROCm과 Docker가 설치된 것을 가정 (ROCm 설치 글 Ubuntu, ROCm 설치 및 제거 (tistory.com)) (Docker 설치 글 Ubuntu, docker 설치 및 제거 정리 (tistory.com)) 공식 Docker image rocm/pytorch - Docker Image | Docker Hub Docker Hub hub.docker.com 1. rocm/pytorch 이미지 run size가 꽤나 크므로 시간이 조금 걸릴 수 있다 sudo docker run -it --network=host --device=/dev/kfd --device=..
Pytorch, grad-cam 사용 정리 Image 분류 모델의 gradcam 사용 정리 pytorch grad cam 공식 깃헙 GitHub - jacobgil/pytorch-grad-cam: Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more. GitHub - jacobgil/pytorch-grad-cam: Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Clas Advanced AI Explainabi..
Ubuntu, docker 설치 및 제거 정리 Ubuntu 20.04 1. Docker 제거 1-1. Package 제거 # Docker 공식 가이드 sudo apt-get purge docker-ce docker-ce-cli containerd.io docker-compose-plugin sudo apt-get autoremove # askubuntu.com 답변 # https://askubuntu.com/questions/935569/how-to-completely-uninstall-docker sudo apt-get purge -y docker-engine docker docker.io docker-ce docker-ce-cli docker-compose-plugin sudo apt-get autoremove -y --purge docker-e..
Ubuntu, TensorRT로 Yolov5 inference Ubuntu 20.04 TensorRT 8.4.3.1 공식 깃헙: GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite. Contribute to ultralytics/yolov5 development by creating an account on GitHub. github.com 공식 깃헙에 따르면, Yolov5를 TensorRT를 통해 배포하는 방법에는 1. export.py를 거쳐 model.engine을 만든 ..
Ubuntu, ROCm, AMD GPU 설치 및 제거 Ubuntu 20.04, 5.15.0-53-generic 에 설치 1. rocm 제거 # 패키지 제거 sudo amdgpu-uninstall sudo apt autoremove amdgpu-dkms sudo apt-get purge amdgpu-install # 파일 제거 sudo rm /etc/apt/sources.list.d/rocm* sudo rm /etc/apt/sources.list.d/amdgpu* # 캐쉬 제거 sudo rm -rf /var/cache/apt/* sudo apt-get clean all 이후 재부팅 2. rocm 설치 2-1. package download sudo apt-get update wget https://repo.radeon.com/amdgpu-install/5.4..