Ubuntu 16.04 + CUDA 8.0 + Theano 深度學習環境搭建

隨著深度學習技術的迅速發展，越來越多的研究者和開發者開始使用各種工具和框架來構建和訓練神經網絡。Theano 是一個流行的深度學習庫，能夠高效地進行數值計算，並且支持 GPU 加速。本文將介紹如何在 Ubuntu 16.04 系統上搭建 CUDA 8.0 和 Theano 的深度學習環境。

環境準備

在開始之前，確保你的系統已經安裝了以下軟件：

• Ubuntu 16.04
• NVIDIA 驅動程序
• CUDA 8.0
• Theano

安裝 NVIDIA 驅動程序

首先，你需要安裝 NVIDIA 驅動程序以支持 GPU 加速。可以通過以下命令安裝：

sudo apt-get update
sudo apt-get install nvidia-375

安裝完成後，重啟系統以使驅動程序生效。

安裝 CUDA 8.0

接下來，下載 CUDA 8.0 的安裝包。你可以從 NVIDIA 的官方網站獲取最新的 CUDA 版本。下載後，使用以下命令進行安裝：

sudo sh cuda_8.0.XX_linux.run

在安裝過程中，選擇安裝驅動程序和 CUDA 工具包。安裝完成後，將 CUDA 的路徑添加到環境變量中：

echo 'export PATH=/usr/local/cuda-8.0/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc

安裝 Theano

接下來，我們需要安裝 Theano。首先，確保你已經安裝了 Python 和 pip。可以使用以下命令安裝：

sudo apt-get install python-pip
sudo pip install --upgrade pip

然後，使用 pip 安裝 Theano：

sudo pip install Theano

配置 Theano

安裝完成後，需要配置 Theano 以使用 CUDA。創建一個名為 .theanorc 的配置文件：

nano ~/.theanorc

在文件中添加以下內容：

[global]
device = cuda
floatX = float32

[nvcc]
fastmath = True

[blas]
ldflags = -lopenblas

測試環境

為了確保一切正常運行，可以創建一個簡單的 Theano 測試腳本：

import theano
import theano.tensor as T

x = T.dscalar('x')
y = T.dscalar('y')
z = x + y

f = theano.function([x, y], z)

print(f(1, 2))  # 應該輸出 3.0

運行這個腳本，如果一切正常，你應該能夠看到正確的輸出。

總結

本文介紹了如何在 Ubuntu 16.04 系統上搭建 CUDA 8.0 和 Theano 的深度學習環境。這一過程包括安裝 NVIDIA 驅動程序、CUDA 工具包以及 Theano，並進行必要的配置。隨著深度學習的普及，擁有一個高效的開發環境將有助於研究者和開發者更好地進行實驗和開發。

如果你需要穩定的 VPS 來支持你的深度學習項目，Server.HK 提供多種選擇，適合不同需求的用戶。