美文网首页
Docker 运行GPU 遇到的问题

Docker 运行GPU 遇到的问题

作者: NazgulSun | 来源:发表于2023-02-23 15:40 被阅读0次
    奇怪的现象

    之前在 Docker 跑pytorch一直没有问题,然后下载了一个paddle 镜像。
    由于paddle镜像与我们自己的显卡 3090 不兼容,所以在docker 内部升级了 cuda 驱动。
    然后所有的 docker 容器内,gpu 都是non_avaiable了。
    在docker 内部执行 nvidia-smi, 能够看到显卡,但是 Cuda version 是N/A

    升级的步骤

    https://zhuanlan.zhihu.com/p/396292888, 基本上是follow 这里面的步骤。

    image.png
    然后这里选择了,安装驱动,460版本。
    因为不选择这个,无法升级下去。

    问题的原因

    因为在docker 内部选择了按照驱动driver 460版本。会影响到物理机,而物理机之前的driver 是高于460的比如470 才能使用。
    所以这个造成了显卡gpu 无法使用。
    最终我们在物理机上升级了显卡驱动解决了问题。

    如何避免docker 中更新driver

    就是手动自己安装。不用连接里提到的方式

    相关文章

      网友评论

          本文标题:Docker 运行GPU 遇到的问题

          本文链接:https://www.haomeiwen.com/subject/dixtldtx.html