개요

PyTorch를 사용할 일이 있어 Torch 설치법을 찾아본 뒤 정리해둔다.
해당방법으로 14.04, 16.04 모두 정상 설치에 성공했다.

설치방법

$ git clone https://github.com/torch/distro
$ cd distro
$ export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__"
$ bash install-deps
$ ./install.sh

참고자료