docker中能安裝cuda;將下載的軟體包上傳到指定的目錄後執行「sudo sh cuda_下載的版本_linux.run」進行安裝即可,cuda是顯示卡廠商NVIDIA退出的運算平台,安裝完成後可利用「nvidia-smi」查看安裝的cuda版本。
本教學操作環境:linux7.3系統、docker19.03版、Dell G3電腦。
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
(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中文網其他相關文章!