Docker 내가 가지고 있는 nvidia GPU의 하위버전이면 모든 환경세팅이 가능하다 내가 사용하는 GPU는 cuda 11.7 version의 nvidia gpu이다, 즉 cuda 11.7 이하의 cuda version의 도커는 모두 사용 가능하다. 해당 docker는 cuda 11.7을 기준으로 설정된 docker image이므로, 만약 하위 버전의 cuda를 사용한다면 google이나 docker hub를 통해 다운받으면 된다. Dockerfile을 내가 작업하고자 하는 디렉토리에 복사한 후, 다음 명령어로 image를 bulid한다. docker build --tag jax_test . 이후, 아래와 같이 쉘 스크립트를 하나 만들어서 실행하면 된다. 스크립트의 파라미터들은 각자의 환경에 맡게 설..

JAX를 이해하기전 꼭 필요한 프레임워크 지식 1. Google Google은 2015년에 기계학습 라이브러리 Tensorflow를 오픈소스로 공개한다. 2. Facebook(+ Microsoft) Facebook은 2016년에 딥러닝 라이브러리 Pytorch를 오픈소스로 공개한다. JAX는 Google에서 개발되었고, 이 말은 XLA가 자유롭게 가능하다는 것이다. 물론 최근(2019년)에 pytorch도 XLA가 가능하게끔 google과 facebook간의 협약이 있었다고 하지만, pytorch를 주로 쓰는 나인데도 XLA라는 용어를 이번에 처음듣게 되었다. 그래서 JAX를 알아보기 전에 XLA를 간단히 알고가려 한다. XLA(=Accelerated Linear Algebra)는 Tensorflow의 ..