[TVM] TVM 설치하기(feat. Anaconda 이용)

 

이번에는 TVM을 설치하는 것을 알아보도록 하겠습니다. 서버에서 tvm다운받으려고 엄청나게 고생했습니다...

 

제 환경은 Ubuntu 18.04 입니다. 

 

TVM을 설치하기 위해서는 LLVM과 cmake를 미리 설치하는것이 필요합니다. 

 

LLVM 설치는 제 포스팅을 보고 따라하시기 바랍니다. 

2021.07.11 - [인턴] - [컴파일러] LLVM 다운로드 및 설치 방법

[컴파일러] LLVM 다운로드 및 설치 방법

 

먼저 자신이 tvm을 설치할 폴더로 이동해 clone을 받습니다.(저는 home 폴더에 다운을 받을 예정입니다)

 

clone 받은 tvm 폴더로 이동해 다음 명령어를 수행합니다. 

 

명령어는 tvm폴더에 build라는 폴더를 만들고(mk) config.cmake 파일을 build 폴더안에 복사하는 것입니다. 그리고 build 폴더로 들어갑니다. 

 

그러면 config.cmake 파일이 생성된것을 알 수 있고 이제 이 파일을 수정할 것입니다. 

저는 다음과 같이 수정하였습니다.

• set(USE_CUDA OFF)를 set(USE_CUDA ON)로 수정했습니다.
• set(USE_GRAPH_EXECUTOR ON) 와 set(USE_PROFILER ON)으로 수정했습니다.
• set(USE_LLVM /경로/llvm/bin/llvm-config)를 추가했습니다. (경로는 자신의 다운받은 llvm경로를 입력하세요)

혹시 더 수정하실 사항이 있으면 TVM 설치가이드를 보고 추가로 수정하시기 바라겠습니다. (마지막에 설치가이드 링크를 추가하겠습니다)

 

이제 tvm폴더에서 이전에 만들었던 build폴더로 들어갑니다. 그리고 다음 명령어를 차례대로 수행합니다. 

 

이제 tvm 설치는 끝났고 python 패키지만 설치하면 됩니다. 기본적으로 python 3.7.0이상에서만 tvm이 설치가 가능했습니다. 

저는 서버에서 다운을 받아야했기 때문에 conda를 사용해서 python 3.7.0 버전을 만들고 다음을 수행했습니다. 

 

conda를 설치받고 새로운 가상환경을 만드는 것은 다음 포스팅을 참고해주세요.

2021.08.21 - [인턴] - [Conda] Conda 설치하기(Feat. Conda 기본 명령어)

 

자신의 홈폴더로 가서 .bashrc를 vim 으로 열어주세요. 

vim .bashrc

그리고 다음과 같이 입력합니다. TVM_HOME이라는 변수에는 자신의 tvm폴더가 있는 위치를 써야합니다.

 

마지막으로 우리가 설치한 tvm폴더안에 python폴더로 들어가 다음 명령어를 수행합니다. 

이제 tvm 설치가 모두 완료되었습니다. 다운받는데 저는 꽤 오랜시간이 걸렸네요... 이글을 보시고 도움이 되셨으면 좋겠습니다. 

 

감사합니다.