美文网首页
docker里找不到gpu

docker里找不到gpu

作者: ccccxxxxcccc | 来源:发表于2019-04-25 17:14 被阅读0次

lspci | grep -i nvidia

可以看到显卡信息

dpkg --list | grep nvidia-*

之后无反应

cat /proc/driver/nvidia/version

可以看到驱动的版本

但是nvidia-smi有问题

目前考虑的问题可能是cuda版本和该驱动版本不一致

该docker里面的cuda版本是8.0

而主机的cuda版本是9.0

docker cuda版本 主机cuda版本

没有什么特别好的解决方法只能重新进一个容器,然后再配一遍环境

sudo nvidia-docker run -t -i bvlc/caffe:gpu /bin/bash

用nvidia-docker命令进入一个新的容器

之后运行nvidia-smi便可以看到gpu信息

这里面默认的cuda版本仍然是8.0

需要安装9.0版本

相关文章

网友评论

      本文标题:docker里找不到gpu

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