2019년 9월 7일 토요일

neural style with CUDA 10.1

Neural Style을 2080ti GPU기반의 컴퓨터에 설치를 우여곡절 끝에 성공했다.

해야할 일은 다음과 같다.
  1. gcc 6 설치 (gcc-7 이상으로 torch 컴파일이 되지 않음.)
  2. cmake  최신으로 설치
  3. torch 패치 (https://medium.com/@jjeaby/cuda-10-torch-%EC%84%A4%EC%B9%98-%EB%B0%A9%EB%B2%95-23fc56d2e58)
  4. torch 소스코드내 CUDA 9이상 지원하지 않는 함수 수정(__all, __shfl, __shfl_xor 등의 함수를 _all_sync, __shfl_sync, __shfl_xor_sync 등 지원되는 함수로 수정)
  5. 설치 완료! (엄청 빠르다)

댓글 없음:

댓글 쓰기