rocm3 Ubuntu, ROCm, AMD GPU, Docker, Tensorflow, 환경에서 JAX 세팅 정리 Ubuntu 22.04 Radeon RX 6800 google-research에서 공개한 maxim 모델을 구동하려니 Tensorflow 기반의 JAX라는 GPU에 dependent한 연산 라이브러리 환경이 필요했다. Docker 기반 ROCm에 동작하는 JAX 환경 구축 정리 도커와 ROCm이 설치된 것을 가정 ROCm 개발자들이 도커에서 JAX를 build를 업데이트 하고 command을 안내하였지만 그대로 돌리면 build가 완료되지 않는다.(https://github.com/ROCmSoftwarePlatform/jax/tree/main/build/rocm) 디버깅 내용을 정리 1. git clone git clone https://github.com/ROCmSoftwarePlatform/jax.g.. 2022. 12. 28. 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=.. 2022. 12. 10. 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.. 2022. 12. 3. 이전 1 다음
