Ubuntu 16.04 CUDA, CUDNN 및 GPU 버전을 설치하는 TensorFlow 일반 절차 요약
1942 단어 tensorflow
카드 드라이버를 설치하는 데는 여러 가지 방법이 있다. 심지어 일부 설정 파일을 명령행 단말기 모드로 바꾸어 조작해야 하는 경우도 있다. 그러나 내가 드라이버를 여러 번 설치한 경험을 보면 나는 지금까지 이렇게 힘든 적이 없었다.Ubuntu 시스템의 System Settings-> Software & Updates -> Additional Drivers에 해당하는 그래픽 카드와 CPU 드라이브를 설치하고, 재부팅 후 System Settings-> Details에서 자신의 그래픽 모델을 볼 수 있다면 드라이브가 성공적으로 설치되었음을 의미하며, 이때 명령줄 아래에 입력
nvidia-smi
해도 그래픽 관련 정보를 볼 수 있다.2. CUDA 설치
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 설치
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. 자주 묻는 질문
sudo gedit /etc/ld.so.conf
에 라이브러리 파일이 있는 경로를 추가한 다음sudo ldconfig
sudo apt-get install/pip3 install **
더 많은 하이라이트를 획득, "seniusen"을 주목 하세요!
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
Mediapipe를 사용한 맞춤형 인간 포즈 분류OpenCV의 도움으로 Mediapipe를 사용하여 사용자 지정 포즈 분류 만들기 Yoga Pose Dataset을 사용하여 사용자 정의 인간 포즈 분류를 생성하겠습니다. 1. 리포지토리 복제: 데이터세트 다운로드:...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.