站长资讯网
最全最丰富的资讯网站

docker能装cuda吗

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

赞(0)
分享到: 更多 (0)
网站地图   沪ICP备18035694号-2    沪公网安备31011702889846号