Neural Style을 2080ti GPU기반의 컴퓨터에 설치를 우여곡절 끝에 성공했다.
- Neural Style: https://github.com/jcjohnson/neural-style
- GPU: 기가바이트 2080 ti
- NVCC: 10.1.243
- Nvidia Driver: 430.40
- CUDA Version: 10.1
- OS: 우분투 18.04
해야할 일은 다음과 같다.
- gcc 6 설치 (gcc-7 이상으로 torch 컴파일이 되지 않음.)
- cmake 최신으로 설치
- torch 패치 (https://medium.com/@jjeaby/cuda-10-torch-%EC%84%A4%EC%B9%98-%EB%B0%A9%EB%B2%95-23fc56d2e58)
- torch 소스코드내 CUDA 9이상 지원하지 않는 함수 수정(__all, __shfl, __shfl_xor 등의 함수를 _all_sync, __shfl_sync, __shfl_xor_sync 등 지원되는 함수로 수정)
- 설치 완료! (엄청 빠르다)
댓글 없음:
댓글 쓰기