实验室的项目采用的是师兄们的祖传代码,牵扯这么多环境,装了几次了,又忘了具体步骤,这次写的记下来。
使用SSH在服务器上安装
1、安装显卡驱动及cuda(今天想试试用命令直接装驱动,之前都是去文本界面去装,觉得太麻烦)
去官网搜索相应的显卡驱动,直接搜索了ubuntu16.04的安装包(之前还试了直接用apt install nvidia-… 没用,安装完了总是报错)
我下载了下面这个包,是T4的显卡
nvidia-diag-driver-local-repo-ubuntu1604-410.129_1.0-1_amd64.deb
*然后按照官网执行命令
*dpkg -i nvidia-diag-driver-local-repo-ubuntu1604-410.129_1.0-1_amd64.deb
*apt-get update
*apt-get install cuda-drivers
*reboot
重启以后,怀着忐忑的心敲下nvidia-smi
出现了以下界面
哈哈,显卡驱动就这么装好了。
去下载对应版本的cuda和cudnn,这两个一般没什么问题,按步骤安装就行。
推荐cuda10.0和cudnn7.4.2。
装完后别忘