docker中能安裝cuda;將下載的軟件包上傳到指定的目錄后執(zhí)行“sudo sh cuda_下載的版本_linux.run”進(jìn)行安裝即可,cuda是顯卡廠商N(yùn)VIDIA退出的運(yùn)算平臺(tái),安裝完成后可以利用“nvidia-smi”查看安裝的cuda版本。
本教程操作環(huán)境:linux7.3系統(tǒng)、docker19.03版、Dell G3電腦。
docker能裝cuda
CUDA(Compute Unified Device Architecture),是顯卡廠商N(yùn)VIDIA推出的運(yùn)算平臺(tái)。 CUDA是一種由NVIDIA推出的通用并行計(jì)算架構(gòu),該架構(gòu)使GPU能夠解決復(fù)雜的計(jì)算問題。 它包含了CUDA指令集架構(gòu)(ISA)以及GPU內(nèi)部的并行計(jì)算引擎。 開發(fā)人員可以使用C語言來為CUDA™架構(gòu)編寫程序,所編寫出的程序可以在支持CUDA™的處理器上以超高性能運(yùn)行。CUDA3.0已經(jīng)開始支持C++和FORTRAN。
示例如下:
docker_Ubuntu16.04下安裝cuda
(1)下載安裝文件。首先去英偉達(dá)官網(wǎng)下載cuda安裝包:https://developer.nvidia.com/cuda-toolkit-archive
(2)更新系統(tǒng)驅(qū)動(dòng)
系統(tǒng)設(shè)置-軟件和更新-附加驅(qū)動(dòng)-更新nvidia和intel
假如更新驅(qū)動(dòng)出現(xiàn)了循環(huán)登錄系統(tǒng)的問題,請(qǐng)先卸載驅(qū)動(dòng),然后使用帶驅(qū)動(dòng)的安裝方法來解決問題(萬能藥)。
此時(shí)驅(qū)動(dòng)的卸載方法:由于登陸進(jìn)入不到圖形用戶界面(GUI),但我們可以進(jìn)入到文本用戶界面(TUI),在登陸界面狀態(tài)下,按Ctrl + Alt + f1,進(jìn)入TUI執(zhí)行
sudo /usr/bin/nvidia-uninstall
然后重啟
sudo reboot
(3)cuda安裝
下載的軟件包上傳到自己定義的目錄后執(zhí)行下面命令
sudo sh cuda_9.0.176_384.81_linux.run
然后用一本書的書角壓住回車鍵,直到服務(wù)條款顯示到100%。接著按下面的步驟選擇
-
accept
-
n(不要安裝driver)
-
y
-
y
-
y
在執(zhí)行上一步時(shí)可能會(huì)報(bào)以下錯(cuò)誤
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
?。?)設(shè)置環(huán)境變量
打開主目錄下的 .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
然后在終端運(yùn)行: cat /usr/local/cuda/version.txt
則顯示:
CUDA Version 9.0.176
cuda安裝成功
推薦學(xué)習(xí):《docker視頻教程》