docker里找不到gpu

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版本