docker中能安裝cuda;將下載的軟件包上傳到指定的目錄后執行“sudo sh cuda_下載的版本_linux.run”進行安裝即可,cuda是顯卡廠商NVIDIA退出的運算平臺,安裝完成后可以利用“nvidia-smi”查看安裝的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
(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視頻教程》