lspci | grep -i nvidia
可以看到显卡信息

dpkg --list | grep nvidia-*
之后无反应
cat /proc/driver/nvidia/version
可以看到驱动的版本

但是nvidia-smi有问题

目前考虑的问题可能是cuda版本和该驱动版本不一致
该docker里面的cuda版本是8.0
而主机的cuda版本是9.0


没有什么特别好的解决方法只能重新进一个容器,然后再配一遍环境
sudo nvidia-docker run -t -i bvlc/caffe:gpu /bin/bash
用nvidia-docker命令进入一个新的容器
之后运行nvidia-smi便可以看到gpu信息

这里面默认的cuda版本仍然是8.0
需要安装9.0版本

网友评论