cuda/tensorflow/keras 설치 노트

1324 단어 기계 학습
  • sudo yum install kernel-devel-$(uname -r),kernel-headers-$(uname -r) 수동 다운로드,rpm -ivh 설치
  • nouveau driver를 비활성화하고 만들기/etc/modprobe.d/blacklist-nouveau.conf:
     
    1 2
     
    blacklist nouveau options nouveau modeset=0
  • Regenerate the kernel initramfs and 리셋 머신: sudo dracut --force
  • sh cuda_8.0.44_linux.run(모든 기본 옵션) & chmod 755 -R /usr/local/cuda-8.0
  • cudnn 설치:
     
    1 2 3 4 5 6 7 8
     
    tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz cd cudnn-8.0-linux-x64-v5.1 sudo cp lib64/*/usr/local/cuda/lib64/sudo cp include/*/usr/local/cuda/include/복사 후 소프트 링크가 효력을 상실할 수 있으므로 소프트 링크를 다시 만들어야 합니다. ln -s libcudn.so.6.0.21 libcudnn.so.6 ln -s libcudnn.so.6 libcudnn.so
  • 환경 변수를 설정합니다.
     
    1 2
     
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64 export PATH="/usr/local/cuda-8.0/bin:$PATH"
  • pip install tensorflow/theano/keras(패키지를 다운로드할 수 없으면 수동으로 다운로드하여 설치)
     
    1
     
    pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.2.0-cp35-cp35m-manylinu
  • 좋은 웹페이지 즐겨찾기