Ubuntu 18.04 深度學習環境配置指南:TensorFlow、Keras、PyTorch 簡易安裝

2021-02-20 AI有道

超微塔式伺服器

顯卡 NVIDIA TITAN Xp *4
內存 128G
CPU 2620V4* 2
電源 1600w *2
硬碟 256G*2+2T*2

二、安裝
1.安裝Ubuntu
使用U盤進行Ubuntu作業系統的安裝:
參考:

https://jingyan.baidu.com/article/a3761b2b66fe141577f9aa51.html
一開始安裝選擇"Install Ubuntu"回車後過一會兒屏幕如果顯示「輸入不支持」,這和Ubuntu對顯卡的支持有關,在安裝主界面的F6,選擇nomodeset,就可以進入下一步安裝了。
安裝過程略,安裝鏡像下載地址:
ubuntu.com/download/des
下載:ubuntu-18.04.2-desktop-amd64.iso

2.安裝ssh
備註:這一步需要到伺服器桌面上的命令窗口輸入,這一步完成後,就可以用ssh工具遠程連接伺服器了,本文使用的是XShell。

sudo apt-get install openssh-server

3. 安裝NVIDIA TITAN Xp顯卡驅動
默認安裝的顯卡驅動不是英偉達的驅動,所以先把舊得驅動刪除掉。

sudo apt-get purge nvidia*

添加Graphic Drivers PPA

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

查看合適的驅動版本:

ubuntu-drivers devices

圖:可用的nvidia 驅動列表

圖中可以看出推薦的是最新的430版本的驅動,安裝該驅動:

sudo apt-get install nvidia-driver-430

安裝完畢後重啟機器:

sudo reboot

重啟完畢運行

nvidia-smi

看看生效的顯卡驅動:

圖:生效的顯卡驅動

4.安裝依賴庫

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-devlibgl1-mesa-glx libglu1-mesa libglu1-mesa-dev


5.GCC降低版本
CUDA9.0要求GCC版本是5.x或者6.x,其他版本不可以,需要自己進行配置,通過以下命令才對GCC版本進行修改。

# 版本安裝:
sudo apt-get install gcc-5
sudo apt-get install g++-5

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 50
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 50


6.安裝Anaconda和tensorflow、keras和pytorch

重點:讓conda自動安裝cuda和cudnn!!!
由於Anaconda可以提供完整的科學計算庫,所以直接使用Anaconda來進行相關的安裝。
1)安裝Anaconda
下載地址:https://www.anaconda.com/download/
這裡我們下載Python 3.7 64bit 的linux版本。
安裝:

bash Anaconda3-2019.03-Linux-x86_64.sh

2)更改pip和conda為國內的源
由於國內訪問pip和conda比較慢,建議更改為國內的源:
a.更改pip的源為阿里云:

mkdir ~/.pip

cat > ~/.pip/pip.conf << EOF
[global]
trusted-host=mirrors.aliyun.com
index-url=https://mirrors.aliyun.com/pypi/simple/
EOF

備註:conda 國內源都封了,不需要更換源了

3)在Anaconda中安裝Python3.7的虛擬環境
創建一個Python的虛擬環境

conda create --name tf python=3.7 #創建tf環境

虛擬環境主要命令:

source activate tf #激活tf環境
source deactivate tf #退出tf環境
conda remove --name tf --all #刪除tf環境(全部刪除)

4)在Anaconda中安裝TensorFlow GPU 1.9

conda install tensorflow-gpu==1.9

將會自動安裝:
cuda,cudnn以及相關的其他組件

5)使用下列代碼測試安裝正確性
命令行輸入:

source activate tf
python

Python命令下輸入以下代碼:

import tensorflowas tf
hello= tf.constant('Hello, TensorFlow!')
sess= tf.Session()
print(sess.run(hello))

沒有報錯就是配置好了。

6)安裝Keras
直接在這個虛擬環境中安裝:

pip install keras

7)安裝Pytorch
直接在這個虛擬環境中安裝:

conda install pytorch torchvision -c pytorch

系統會自動安裝cuda和cudnn
測試Pytorch是否安裝成功:
命令行輸入:

source activate tf
python

python命令下輸入以下代碼:

import torch
print(torch.cuda.is_available())

返回True說明安裝成功了。


總結
為避免讀者踩坑,本文測試成功了Ubuntu18.04環境下配置深度學習環境包括:CUDA+CUDNN+TensorFlow1.9+Pytorch1.1的安裝和測試。

參考
http://blog.csdn.net/weixin_41863685/article/details/80303963

覺得這篇文章有幫助?請轉發給更多人

關注 AI有道 加星標,獲取最新 AI 乾貨

最新 AI 乾貨,我在看❤️

相關焦點

  • 深度學習環境配置指南:Pytorch、TensorFlow、Keras
    導讀本文介紹了作者使用RTX3090進行深度學習環境配置pytorch、tensorflow、keras等的詳細過程及代碼。筆者中山大學研究生,醫學生+計科學生的集合體,機器學習愛好者。最近剛入了3090,發現網上寫的各種環境配置相當混亂而且速度很慢。所以自己測了下速度最快的3090配置環境,歡迎補充!
  • RTX 3090 的深度學習環境配置指南:Pytorch、TensorFlow、Keras
    導讀本文介紹了作者使用RTX3090進行深度學習環境配置pytorch、tensorflow、keras等的詳細過程及代碼。筆者中山大學研究生,醫學生+計科學生的集合體,機器學習愛好者。最近剛入了3090,發現網上寫的各種環境配置相當混亂而且速度很慢。所以自己測了下速度最快的3090配置環境,歡迎補充!
  • RTX 3090的深度學習環境配置指南:Pytorch、TensorFlow、Keras
    最近剛入了3090,發現網上寫的各種環境配置相當混亂而且速度很慢。所以自己測了下速度最快的3090配置環境,歡迎補充!(1)官網下載,安裝顯卡驅動:bash NVIDIA-Linux-x86_64-455.23.04.run(2)安裝Anaconda並換源
  • 深度學習筆記15:ubuntu16.04 下深度學習開發環境搭建與配置
    作者:魯偉一個數據科學踐行者的學習日記。
  • 從零開始深度學習第15講:ubuntu16.04 下深度學習開發環境搭建與配置
    深度學習大多用來處理圖片、語言、視頻和文本等大量非結構化數據,因而計算資源是進行深度學習開發的必要條件,想依靠windows單機玩深度學習恐怕不現實,基於linux和 gpu伺服器的深度學習開發環境配置就顯得尤為重要了。
  • Ubuntu 18.04安裝Tensorflow(CPU)
    最近我開始學習深度學習框架Tensorflow,一開始在Windows平臺下的anaconda下安裝,由於anaconda安裝幾次後navigator
  • 【AI學習篇】Keras+Tensorflow: GPU加速環境搭建與測試
    令人欣喜的是,大多數Nvidia顯卡都可以使用Cuda這一顯卡加速運算平臺(詳見:https://developer.nvidia.com/cuda-gpus),使得我們許多預算有限的童鞋可以有機會了解和熟悉深度學習框架。本文薈薈帶領大家一起配置好gpu加速的tensorflow環境,並使用smallvggnet這一典型的深度卷積神經網絡對環境進行測試。
  • 深度學習環境配置 (Ubuntu18.04 + CUDA10.0 + cuDNN7.6.5 + TensorFlow2.0)
    比如裝好了 CUDA 重啟後進不去桌面系統了,直接黑屏、比如滑鼠鍵盤都不 work 了、再比如裝好了卻安裝不了 TensorFlow-GPU.看了一圈網上的安裝教程,發現還是官方指南真香了~新年第一篇,分享一下我的 Ubuntu 18.04 + CUDA 10.0 + cuDNN 7.6.5 + TensorFlow 2.0
  • 手把手教你安裝深度學習軟體環境(附代碼)
    本文向你解釋如何在一臺新裝的 Ubuntu 機器上安裝 Python 和 Nvidia 硬體驅動、各類庫和軟體包。為了進行強化學習研究,我最近購置了一臺基於 Ubuntu 和英偉達 GPU 的深度學習機器。儘管目前在網絡中能找到一些環境部署指南,但目前仍然沒有全面的安裝說明。另外,我也不得不閱讀了很多文檔來試圖理解安裝細節——其中的一些並不完整,甚至包含語法錯誤。
  • 如何在 Ubuntu 16.04 上安裝並使用 TensorFlow
    連結:https://www.digitalocean.com/community/tutorials/how-to-install-and-use-tensorflow-on-ubuntu-16-04?
  • Ubuntu 16.04搭建深度學習環境的超級無敵詳細教程
    Ubuntu 16.04搭建深度學習環境前言    Linux伺服器上的環境沒有共用,要在自己文件夾搭建所需要的環境。要安裝的有:Anaconda3、TensorFlow、Pytorch,這幾個都是直接或間接的有版本對應關係。順便記錄一下安裝過程,本來搭環境就很煩,以後在新機器搭環境就不用東找西找。
  • 2021 年了,TensorFlow 和 PyTorch 兩個深度學習框架地位又有什麼變化嗎?
    思想火龍果回答:pytorch統治學術圈。現在各大頂會開原始碼沒有一絲絲的tensorflow2.x,就連谷歌家自己的論文也是只有torch和jax。零零星星一些tf1的開原始碼。tensorflow2.x難用嗎?真不難用的,基本就是在用keras,tf.data pipeline 和 tf.distribute 用起來真的省心又快。
  • 從零開始搭建深度學習伺服器:TensorFlow + PyTorch + Torch
    本教程轉載於深度學習中文社區(studydl.com).在深度學習大行其道的今天,我們不應該停留於觀望的階段,我們應該多多上手進行實踐,下面將為大家介紹一下最簡單也是最基礎的內容,配置一個自己的深度學習伺服器.1.
  • Windows下tensorflow-gpu=1.13.2、torch1.2.0深度學習環境配置
    需要特別注意的問題是版本的對應問題和安裝順序問題一般情況下要考慮的有 tensorflow(或 pytorch)、 cuda、cudnn的版本對應,有時候還需考慮 python4、CUDA和CUDNN下載5、配置環境
  • RTX 3080深度學習環境配置
    找資料的時候感覺現在寫30XX系顯卡深度學習環境配置的文章還不太完善,所以記錄了自己的採坑經歷
  • Keras,TensorFlow 的高級 API
    tf.keras 是 TensorFlow 的高級 API,旨在構建和訓練深度學習模型。此 API 可用於快速原型設計、尖端研究和實際生產,並具備三項關鍵優勢:簡單易用Keras 界面簡單一致,已針對常見用例進行優化。
  • TensorFlow2.0學習筆記
    安裝Anconda4.1. Win10安裝Anconda4.2. Linux安裝Anconda五. TensorFlow2.0正式版安裝5.1. 首先配置清華鏡像源5.2. 安裝TensorFlow2.0六. TensorFlow2.0實戰Mnist數據集6.1.
  • Python深度學習研究開發環境安裝--TensorFlow與PyTorch
    Anaconda發布版本,並提供了Anaconda和深度學習開源軟體平臺TensorFlow和PyTorch的安裝方法。一、機器學習開源軟體框架發展動態我們知道,機器學習開源軟體框架對機器學習的研究與廣泛應用起到了積極的推動作用,大大降低了進入深度學習應用研究領域的技術門檻和難度。對於機器學習科研工作者和工業界從業人員來說,熟練掌握一種機器學習框架是必備技能之一。隨著深度學習技術發展的突飛猛進和應用的普及,機器學習框架市場也漸漸度過了初期野蠻生長的階段。
  • Windows電腦深度學習環境超詳細配置指南
    要在 Windows 上開發模型,首先當然是配置開發環境。Kaggle Master 及機器學習實踐者 Abhinand 立足於自己的實踐,給出了一種簡單易行的 Windows 深度學習環境配置流程。本文將介紹在 Windows 計算機上配置深度學習環境的全過程,其中涉及安裝所需的工具和驅動軟體。出人意料的是,即便只是配置深度學習環境,任務也不輕鬆。你很有可能在這個過程中犯錯。
  • 9步教你安裝完成tensorflow與pytorch
    配置鏡像源,打開後,可以看到現在是base環境,就是安裝完anaconda之後的自帶環境(⾸先進⾏鏡像源的配置,因為咱們⼤部分⼩夥伴在國內,但是tensorflow和pytorch卻在國外,為了防⽌下載的時候出現⽹絡問題,所以需要配置