首页  >  文章  >  运维  >  docker能装cuda吗

docker能装cuda吗

WBOY
WBOY原创
2022-07-22 16:00:265257浏览

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

22.png

(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中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn