wsl22 WSL2, Nvidia GPU, CUDA 환경에서 super-resolution 모델 (SwinIR) 돌려보기 시스템 사양 (환경) CPU - i7-10700F Memory - 32GB Mainboard - MSI, B460M GPU - Geforce RTX 3060 Ti WSL - Ubuntu 20.04 WSL2 환경에서 Nvidia gpu를 (3060ti) 사용하여 SwinIR 이라는 모델로 super-resolution을 돌려보았다. SwinIR은 이전에 리뷰했던 super-resolution 모델과 달리 transformer 기반이라 따로 시도해 보았다. 먼저 AMD GPU를 이용한 DirectML로 구현을 해보고자 하였으나, DirectML의 pytorch에 attention 관련 함수들이 아직 완벽하게 구현되지 않아 코드를 돌릴수 없었던 관계로 결국, CUDA를 사용하였다. https://github... 2022. 7. 29. 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. 이전 1 다음
