首頁  >  文章  >  運維  >  docker能裝cuda嗎

docker能裝cuda嗎

WBOY
WBOY原創
2022-07-22 16:00:265258瀏覽

docker中能安裝cuda;將下載的軟體包上傳到指定的目錄後執行「sudo sh cuda_下載的版本_linux.run」進行安裝即可,cuda是顯示卡廠商NVIDIA退出的運算平台,安裝完成後可利用「nvidia-smi」查看安裝的cuda版本。

docker能裝cuda嗎

本教學操作環境:linux7.3系統、docker19.03版、Dell G3電腦。

docker能裝cuda

CUDA(Compute Unified Device Architecture),是顯示卡廠商NVIDIA推出的運算平台。 CUDA是一種由NVIDIA推出的通用平行運算架構,該架構使GPU能夠解決複雜的運算問題。它包含了CUDA指令集架構(ISA)以及GPU內部的平行運算引擎。開發人員可以使用C語言來為CUDA™架構編寫程序,所編寫出的程序可以在支援CUDA™的處理器上以超高效能運作。 CUDA3.0已經開始支援C 和FORTRAN。

範例如下:

docker_Ubuntu16.04下方安裝cuda

(1)下載安裝檔。首先到英偉達官網下載cuda安裝套件:https://developer.nvidia.com/cuda-toolkit-archive

docker能裝cuda嗎

(2)更新系統驅動程式

  系統設定-軟體和更新-附加驅動-更新nvidia和intel

    假如更新驅動出現了循環登入系統的問題,請先卸載驅動,然後使用帶驅動的安裝方法來解決問題(萬用藥)。

     此時驅動的卸載方法:由於登陸進入不到圖形使用者介面(GUI),但我們可以進入到文字使用者介面(TUI),在登陸介面狀態下,按Ctrl Alt f1,進入TUI執行

sudo /usr/bin/nvidia-uninstall

     然後重新啟動

sudo reboot

(3)cuda安裝

  下載的軟體包上傳到自己定義的目錄後執行下面命令

sudo sh cuda_9.0.176_384.81_linux.run

  然後用一本書的書角壓住回車鍵,直到服務條款顯示到100%。接著按下面的步驟選擇

  •   accept

  •   n(不要安裝driver)

  • #  y

  •   y

  • #    y

  執行上一個步驟時可能會回報下列錯誤

failed--compilation aborted at ./cuda-installer.pl line 5

  解決方法:如下

sudo apt install freeglut3-dev build-essential libx11-dev libxmu-dev 
libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev libglfw3-dev libgles2-mesa-dev

  (4)設定環境變數

     開啟主目錄下的.bashrc檔案加入以下路徑,例如我的.bashrc檔案在/home/wangyuanwei下,如果沒有找到,則按Ctrl H鍵顯示隱藏檔案。

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0
    然后在终端运行:
cat /usr/local/cuda/version.txt

   則顯示:

CUDA Version 9.0.176

  cuda安裝成功

推薦學習:《docker影片教學

以上是docker能裝cuda嗎的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn