首頁 > 軟體

Ubuntu 16.04 LTS 安裝Theano設定GPU

2020-06-16 17:35:32

之前經過一番折騰,終於在Win8.1下安裝了Theano並設定好GPU(詳見上一篇文章 http://www.linuxidc.com/Linux/2016-08/134143.htm),近幾天又經歷重重磨難,在Unbutn16.04lts下安裝了Theano並設定GPU。

首先列示我的電腦使用軟硬體設定:Ubuntu 16.04 LTS 64位元系統,NVIDIA Geforce 840M,Anaconda python 3.5,Cuda7.5。

在詳述之前,首先記錄一下之前安裝經歷的兩個坑:

1. 我的電腦之前使用的是Ubuntu 14.04 LTS,在該系統上安裝Theano及設定Cuda7.5很容易,按照這篇文章 http://www.linuxidc.com/Linux/2016-08/134144.htm  所寫的方法很快就安裝好了,然而該系統卻有一個嚴重的問題:安裝好Cuda後重新啟動電腦,始終停留在logo介面,無法進入系統。據知乎上所說,這個問題是由於nvidia的驅動和Ubuntu的acpi模組衝突所以造成卡死,而且無法解決,無奈之下只得更換Ubuntu 16.04 LTS。

2. 這一條就是使用Ubuntu 16.04 LTS安裝Cuda所存在的問題。按照NVIDIA官網上提供的方法sudo dpkg -i cuda*.deb, sudo apt-get update, sudo apt-get install cuda,在apt-get update這一步,出現了Release無法下載,後面也就無法進行了,嘗試了幾個方法,也未能解決。後來嘗試了其他方法安裝Cuda,設定GPU成功,才發現,在Ubuntu下設定Theano確實容易,只是由於系統和軟體的相容性問題,導致這麼多么蛾子。最後我又把系統刪除,重新安裝系統及theano,確保下面所述方法無誤。

下面正式介紹在Ubuntu 16.04 LTS系統下安裝Theano及設定GPU的過程。

1. 更新NVIDIA顯示卡驅動

安裝好系統之後首先在系統更新管理器中更新顯示卡驅動,如下圖

點選Apply Changes

2. 安裝Anaconda

在Anaconda官網選擇合適的安裝包下載之後,執行如下命令
  bash Anaconda3-4.1.1-MacOSX-x86_64.sh
安裝好之後重新啟動電腦,本應該在第一步重新啟動電腦,由於我這兩步是同時進行的(因為第一步更新時間較長),所以在這裡重新啟動電腦,但是重新啟動電腦之後驚奇發現,Anaconda的gcc版本自動降為4.4,正好解決了Cuda不支援5.0以上gcc版本的問題,不需要再進行gcc降級了。

但是只檢視gcc –version的話,其版本依然是5.4。

此處不知為何原因,但是後面可以成功設定GPU。

3. 安裝Cuda7.5

直接 sudo apt-get install nvidia-cuda-toolkit 即可安裝Cuda7.5,安裝完成之後可以通過nvcc -V檢視Cuda版本,

而且這種安裝的方便之處在於不需要設定Cuda的環境變數,據知乎上呂俊奇回答“因為安裝在/usr/lib/x86_64-linux-gnu/中,沒有新建資料夾所以不用新增環境變數”。

4. 安裝theano

Anaconda安裝theano可用conda直接安裝
  conda install theano

5. 設定.theanorc

生成檔案 sudo gedit ~/.theanorc(注意不要漏掉theano前面的一個點)並複製如下內容,然後儲存,其中cuda一項的內容為cuda所安裝的位置。
  [global]
  floatX=float32
  device=gpu
  [cuda]
  root=/usr/lib/nvidia-cuda-toolkit
  [nvcc]
  flags=-D_FORCE_INLINES

至此theano的安裝及設定已經完成,在python裡輸入import theano,顯示使用的GPU裝置表示GPU設定成功。

也可以採用Using the GPU中的測試程式對GPU進行測試。

Ubuntu下安裝Torch並設定CUDA和cuDNN  http://www.linuxidc.com/Linux/2016-07/133208.htm 

Ubuntu 16.04系統下CUDA7.5設定Caffe教學  http://www.linuxidc.com/Linux/2016-07/132859.htm

Ubuntu 14.04安裝Nvidia CUDA7.5並搭建Python Theano深度學習開發環境  http://www.linuxidc.com/Linux/2015-09/123562.htm

Ubuntu下CUDA(含GPU卡驅動)安裝過程  http://www.linuxidc.com/Linux/2015-07/120456.htm

Caffe + Ubuntu 14.04 64bit + CUDA 6.5 設定說明  http://www.linuxidc.com/Linux/2015-04/116444.htm

Ubuntu 14.04 安裝設定CUDA  http://www.linuxidc.com/Linux/2014-10/107501.htm

Ubuntu 12.04設定NVIDIA CUDA5.5實錄  http://www.linuxidc.com/Linux/2014-10/107502.htm

Ubuntu安裝Theano+CUDA  http://www.linuxidc.com/Linux/2014-10/107503.htm

本文永久更新連結地址http://www.linuxidc.com/Linux/2016-08/134145.htm


IT145.com E-mail:sddin#qq.com