728x90
deep learning 을 위해 torch 를 사용하는 경우는 보통 cuda 8, cudnn 5 를 설치해서 사용하곤 한다. 하지만, 다른 deep learning framework 에서는 cuda 8 를 지원하지 않아 cuda 9을 설치해야 하는 경우가 생긴다.
그래서, cuda 9 을 설치하고 torch 를 설치하려고 하면 아래와 같은 compile error 가 발생하게 된다.
이런 compile error 가 발생하는 경우 아래와 같은 명령어로 해결이 가능하다.
cd ~/torch; bash install-deps;
./clean.sh
export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__"
./install.sh
