接到一个项目, 要用GPU服务器放一个图像识别的服务, 刚租了阿里的服务器, 需要装一下显卡驱动, 在此记录一下安装步骤。
1 查看机器GPU型号
用一下命令查看NVIDIA的GPU型号,虽然租用服务器时有型号选择,在此还是放一下查看型号的方法
lspci | grep NVIDIA # 查看NVIDIA显卡型号
lspci | grep VGA # 查看集成显卡驱动
这台机器的显卡是Tesla T4
知道型号以后就去NVIDIA官网找驱动吧
2 下载NVIDIA显卡驱动
到nvidia官网, 选择自己需要的驱动程序,为了和开发服务器的环境一致,这里选择cuda10.0, 选好后点击搜索
然后就找到了这个, 点击下载
到这儿, 右击下载, 复制链接地址(直接下载到服务器上吧, 不然还得传上去, 麻烦。。)
到服务器上用wget命令下载就可以了
wget https://cn.download.nvidia.com/tesla/410.129/NVIDIA-Linux-x86_64-410.129-diagnostic.run
(105M的东西, 下载的好慢 >_<)
3 安装NVIDIA驱动
下载好以后,先给这个.run 文件赋予执行权限。
chmod a+x NVIDIA-Linux-x86_64-410.129-diagnostic.run
然后执行安装
sh NVIDIA-Linux-x86_64-410.129-diagnostic.run
完成后输入 nvidia-smi
验证一下。
//TODO 待更新