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视频教程》