Ubuntu 16.04 CUDA, CUDNN 및 GPU 버전을 설치하는 TensorFlow 일반 절차 요약

1942 단어 tensorflow
1. 그래픽 드라이브 설치
카드 드라이버를 설치하는 데는 여러 가지 방법이 있다. 심지어 일부 설정 파일을 명령행 단말기 모드로 바꾸어 조작해야 하는 경우도 있다. 그러나 내가 드라이버를 여러 번 설치한 경험을 보면 나는 지금까지 이렇게 힘든 적이 없었다.Ubuntu 시스템의 System Settings-> Software & Updates -> Additional Drivers에 해당하는 그래픽 카드와 CPU 드라이브를 설치하고, 재부팅 후 System Settings-> Details에서 자신의 그래픽 모델을 볼 수 있다면 드라이브가 성공적으로 설치되었음을 의미하며, 이때 명령줄 아래에 입력nvidia-smi해도 그래픽 관련 정보를 볼 수 있다.
2. CUDA 설치
  • CUDA 홈페이지에서 적합한 버전의runfile 파일을 다운로드한 후 터미널에서 실행sudo sh cuda_filename.run하면 드라이브를 설치할지 물어볼 때 No(위에 설치되어 있음)를 선택하는 것 외에 Yes를 선택할 수 있습니다.
  • 홈 디렉터리에서 터미널 실행sudo gedit .bashrc을 열고 파일 끝에 해당하는 경로를 추가합니다.
  • export PATH=/usr/local/cuda-8.0/bin:$PATH
    export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PA
  • 터미널 실행source .bashrc
  • 단말기 운행nvcc -V, 관련 CUDA 버전 정보 설명 설치가 나타납니다.

  • 3. CUDNN 설치
  • 해당 버전의 CUDNN 압축 패키지를 풀고 해당 파일을 관련 경로로 복사합니다.
  • sudo cp cuda/include/cudnn.h /usr/local/cuda/include
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
    sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

    4. Python 종속 라이브러리 및 TensorFlow 설치
    Ubuntu의 기본 Python 버전은 2.7이기 때문에 시스템의 기본 Python 버전을 바꾸면 소프트웨어를 설치할 때 의존 문제가 발생할 수 있습니다. 현재 주류 Python은 모두 3.x버전이므로 파이톤3을 사용하면서 시스템의 기본 파이톤버전을 바꾸지 않기 위해 pip로 설치하는 것이 좋습니다.
    sudo apt-get install python3-pip
    sudo pip3 install numpy,jupyter,matplotlib
    sudo pip3 install tensorflow-gpu
                      
    sudo pip3 install -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com numpy(  python  )

    5. 자주 묻는 질문
  • *를 찾을 수 없습니다.so 파일, 먼저 라이브러리 파일이 있는 경로를 검색하고 터미널 실행sudo gedit /etc/ld.so.conf 에 라이브러리 파일이 있는 경로를 추가한 다음sudo ldconfig
  • No module named ** , sudo apt-get install/pip3 install **

  • 더 많은 하이라이트를 획득, "seniusen"을 주목 하세요!

    좋은 웹페이지 즐겨찾기