lspci | grep -i nvidia
可以看到显卡信息
![](https://img.haomeiwen.com/i4429433/850dd89adfbfd945.png)
dpkg --list | grep nvidia-*
之后无反应
cat /proc/driver/nvidia/version
可以看到驱动的版本
![](https://img.haomeiwen.com/i4429433/d0607561c288c410.png)
但是nvidia-smi有问题
![](https://img.haomeiwen.com/i4429433/f53233f4e1387ba8.png)
目前考虑的问题可能是cuda版本和该驱动版本不一致
该docker里面的cuda版本是8.0
而主机的cuda版本是9.0
![](https://img.haomeiwen.com/i4429433/5d751498bebd52b2.png)
![](https://img.haomeiwen.com/i4429433/e0be7488472adcde.png)
没有什么特别好的解决方法只能重新进一个容器,然后再配一遍环境
sudo nvidia-docker run -t -i bvlc/caffe:gpu /bin/bash
用nvidia-docker命令进入一个新的容器
之后运行nvidia-smi便可以看到gpu信息
![](https://img.haomeiwen.com/i4429433/c8d9c0f2690c5eca.png)
这里面默认的cuda版本仍然是8.0
需要安装9.0版本
![](https://img.haomeiwen.com/i4429433/dbfd715d22f292b9.png)
网友评论