torch with cuda9 (1) 썸네일형 리스트형 torch cuda9 install error 해결 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 이전 1 다음