Instalacja Tensorflow r0.12 CUDA 8 na ubuntu 16.04

Z racji zakupu nowego laptopa, byłem zmuszony od nowa zainstalować TensorFlow, moje ostatnie doświadczenie były raczej trudne, lecz obecna wersja biblioteki jest już dojrzalsza, to samo można powiedzieć o sterownikach od NVidia. Ostatecznie wszystko przebiegło bardzo sprawnie.

W instrukcji dowiesz się jak zainstalować TensorFlow r0.12 z CUDA Toolkit 8.0 na ubuntu 16.04, można to traktować jako zaktualizowaną wersję poprzedniego wpisu, no może bez takiego dramatyzmu jak ostatnio (http://ksopyla.com/2016/07/instalacja-tensorflow-0-9-cuda-7-5-na-ubuntu-16-04/). Całość instaluję na laptopie Dell Vostro 5486 z dwoma kartami graficznymi wbudowaną Intel HD 620 oraz dedykowaną NVidia Geforce 940MX.

Krok pierwszy –  Instalacja sterowników do NVDIA 375.20

Z racji tego, że instalacja przebiegała na laptopie z dwoma kartami, instalując sterowniki do karty musiałem pozwolić na „Insecure boot”. Po pierwszym restarcie nie uruchomiło mi się Unity w Ubuntu, lecz po kolejnych wszystko było już ok. Po instalacji sterowników należy sprawdzić jaką kartę mamy w danym momencie wybraną, czy wbudowaną czy GeForce, do szczegółów odsyłam do dwóch wpisów:

Po instalacji powinniśmy móc przełączać się pomiędzy kartami z wykorzystaniem poleceń:

Po instalacji powinniśmy mieć zainstalowane narzędzie nvidia-smi

Krok drugi – Instalacja CUDA Toolkit

Ściągamy CUDA Toolkit 8, na stronie NVidia wybieramy kolejno:

  • system – Linux
  • architekture – x86_64
  • dystrybucję – Ubuntu
  • wersję – 16.04
  • typ instalatora – runfile(local) – Uwaga plik zajmuje ~1.5GB.

Nadajemy uprawnienia do wykonywania i uruchamiamy z konsoli

W trakcie instalacji na ekranie zostaniemy zapytani o następujące informacje:

  • Do you accept the previously read EULA?
    accept
  • Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 367.48?
    n (sterowniki mamy zainstalowane już wcześniej)
  • Install the CUDA 8.0 Toolkit?
    y
  • Enter Toolkit Location:
    /usr/local/cuda-8.0 (enter)
  • Do you wish to run the installation with ‚sudo’?
    y
  • Do you want to install a symbolic link at /usr/local/cuda?
    y (ułatwi to nam późniejszą konfigurację)
  • Install the CUDA 8.0 Samples?
    y (ja dałem yes, lecz to czy chcesz mieć przykłady zależy od ciebie i wielkości twojego dysku)
  • Enter CUDA Samples Location:
    enter (domyślnie w katalogu domowym użytkownika)

Po kilku minutach w konsoli otrzymamy takie podsumowanie, nie przejmujcie się warningiem:

Krok trzeci – cuDNN

Tensorflow wymaga biblioteki cuDNN v5 (https://developer.nvidia.com/cudnn), aby ją ściągnąć należy się zarejestrować i zalogować do NVidia developer. Ja ściągnąłem

Następnie kopiujemy pliki do /usr/local/cuda

Krok czwarty – instalacja TensorFlow w virtualenv

 

Na koniec aby wszystko przetestować, uruchomiłem prosty skrypt

W wyniku powinniśmy zobaczyć:

Dwójka na samym końcu jest wynikiem naszego skryptu. Ważne aby udało się załadować wszystkie biblioteki CUDA.

 

4 Comments Instalacja Tensorflow r0.12 CUDA 8 na ubuntu 16.04

  1. Pingback: Instalacja Tensorflow 0.9 CUDA 7.5 na ubuntu 16.04 i diabeł na ramieniu - About Data

  2. Kuba

    Dzięki za tutorial, przydał się i działa 🙂

    Warto tylko jeszcze napisać na końcu o eksporcie dwóch zmiennych (bez tego u mnie python nie wiedział gdzie szukać cuda):

    export LD_LIBRARY_PATH=”$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64″
    export CUDA_HOME=/usr/local/cuda

    Reply
  3. Pingback: Jak przygotować środowisko do Tensorflow z CUDA w virtualenv i Python 3 - About Data

Dodaj komentarz

Twój adres email nie zostanie opublikowany. Pola, których wypełnienie jest wymagane, są oznaczone symbolem *