conda로 설치하기 h2o4gpu

2019. 5. 14. 13:49꿀팁 분석 환경 설정

728x90

https://github.com/h2oai/h2o4gpu

 

h2oai/h2o4gpu

H2Oai GPU Edition. Contribute to h2oai/h2o4gpu development by creating an account on GitHub.

github.com

 

저는 cuda9 로 진행을 하였습니다.

export CUDA_HOME=/usr/local/cuda # or choose /usr/local/cuda9 for cuda9 and /usr/local/cuda8 for cuda8
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_HOME/lib64/:$CUDA_HOME/lib/:$CUDA_HOME/extras/CUPTI/lib64

sudo apt-get install libopenblas-dev pbzip2
sudo apt-get -y install libcurl4-openssl-dev libssl-dev libxml2-dev

기존의 pip가 잘 안되서 환경을 그냥 만들어서 진행

자동으로 python3.6이 됨
conda create -n h2o4gpu -c h2oai -c conda-forge h2o4gpu-cuda9 

source activate h2o4gpu

## jupyter에 커널 추가하기
pip install ipykernel

python -m ipykernel install --user --name h2o4gpu --display-name h2o4gpu


## 주의!!! 꼭 해야하는 작업이 있다.
pip install jupyter==1.0.0
pip install tornado==4.5.3
이걸 해야 jupyter에서 kernel이 폭탄이 되지 않음

 

simple example

 

https://anaconda.org/h2oai/h2o4gpu-cuda9

728x90