DirectML3 WSL2, AMD GPU, DirectML 환경에서 super-resolution 모델 (tecogan) 돌려보기 시스템 사양 (환경) CPU - 3700X Memory - 32GB Mainboard - Msi, MAG B550M MORTAR GPU - RX 6800 (reference) WSL - Ubuntu 20.04 super-resolution 모델 중에서 tecogan 이라는 모델을 구현해 보자 이전 BSRGAN 이 pytorch 기반이었다면, 이번 모델은 tensorflow 기반 모델이라서 따로 한번 더 정리하였다. 정식 깃헙은 아래와 같다. GitHub - thunil/TecoGAN: This repo contains source code and materials for the TEmporally COherent GAN SIGGRAPH project. GitHub - thunil/TecoGAN: This.. 2022. 7. 25. WSL2, AMD GPU, DirectML 환경에서 super-resolution 모델 (BSRGAN) 돌려보기 시스템 사양 (환경) CPU - 5900X Memory - 96GB Mainboard - Gigabyte, X570 Aorus Master GPU - RX 6900XT (reference) WSL - Ubuntu 20.04 WSL2 환경에서 AMD GPU와 (6900XT) DirectML을 활용하여 Super-resoltion 모델을 돌려보았다. Super-resolution 모델이란 닳고 닳은 이미지를 딥러닝 기술을 활용하여 새 이미지처럼 만들어 주는 것이다. 많은 모델이 개발되어 공유되고 있지만 AMD GPU와 DirectML이라는 제한된 환경 탓에 패키지 버전 이슈가 없는 모델로 찾고 찾아 BSRGAN 이라는 모델을 선택하였다. 모델에 대한 자세한 설명은 아래 깃헙 링크에 있다. GitHub - c.. 2022. 7. 18. WSL2, AMD GPU, DirectML 환경에서 Yolov4 돌려보기 시스템 사양 (환경) CPU - 3700X Memory - 64GB Mainboard - Msi, MAG B550M MORTAR GPU - RX 6800 (reference) WSL - Ubuntu 20.04 WSL2 에서 AMD GPU (RX 6800)을 사용해, Yolov4를 돌려보고자 한다. 영상 딥러닝을 하려는 경우 그래픽 카드는 필수적이다. Nvidia와 AMD가 대표적인 그래픽 카드 제조사인데, 딥러닝을 위해서는 서로 다른 방법이 필요하다. Nvidia 그래픽카드의 경우 고민할것 없다. 그냥 CUDA와 cudnn 버전 잘 맞춰서 설치하고 돌리면 된다. (WSL에서도 가능한 것으로 알고있다. Enable NVIDIA CUDA on WSL 2 | Microsoft Docs) AMD 그래픽 카드의 .. 2022. 7. 11. 이전 1 다음