banana like sunshine RSS FEED

GPU 이용은 클라우드로만 하다가, GPU를 득템하게 되어 리눅스 머신에 세팅.

Machine Learning으로 가는 가장 큰 난관은 GPU 잡아주고 드라이버, CUDA 설치하는 거라고 누가 그랬던가..

결국 열 번 정도 OS부터 잘근잘근 다시 깔며 고생. 인터넷에 여기저기 떠도는 팁들은 다 제각각이라 도움이 안되고, NVIDIA에서 제공하는 가이드를 정독하는게 정답입니다.

제가 남겨놓고 싶은 팁 하나. kernel driver와 CUDA driver 사이에 버전 차이 등 kernel 레벨에서 발생하는 문제는 아래 커맨드로 확인하면 도움이 됩니다. 저도 버전 문제로 엄청 헤매다가 어느 고수가 설명도 없이 "이거 한번 해봐"하는 걸 보고 해결했습니다.

sudo dmesg | grep NVRM

dmesg는 kernel ring buffer를 확인해주는 커맨드라고 합니다. 저는 이렇게 하니 client module이랑 kernel module이 안맞다는 사실을 알고 버전만 맞춰주니 바로 해결이 되더군요. ^^

ndarray를 간단하게 shuffle할 때 씁니다:

np.random.seed(seed)
indices = np.arange(len(x_train))
np.random.shuffle(indices)
x_train = x_train[indices]