版权声明:原创作品,欢迎转发!转发附上链接 https://blog.csdn.net/qq_26369907/article/details/88430836
实验室做GPU实验,需要用到NVIDIA显卡+ CUDA10,安装过程中走了很多坑,在此记录。
- 查看本地显卡
XXXXX:~$ lspci | grep VGA
01:00.0 VGA compatible controller: NVIDIA Corporation GP104M [GeForce GTX 1070 Mobile] (rev a1)
可以看到我的显卡是:GeForce GTX 1070的, 如果没有说明显卡没插好或者其他问题,先解决。
- 查看显卡驱动
xxxx:~$ lshw -c video # 看configurure字段有木有driver字样,若有内容,则表示对应的工作站显卡驱动装好了。
## 但是!!! Ubuntu18.04 或者其他很有可能是 nouveau ,此次实验室做深度学习,要用到NVIDIA的,因此需要改显卡驱动,先删除旧的,再禁用nouveau;
2.1 删除原有驱动
sudo apt-get remove --purge nvidia* # 其实这也不删问题也不大
2.2 禁用nouveau驱动
修改blacklist.conf文件,添加 blacklist nouveau,具体操作如下:
sudo vim /etc/modprobe.d/blacklist.conf
blacklist nouveau #添加数据
- 选择合适显卡,安装显卡
3.1 下载显卡驱动
建议前往官网下载,选择对应型号的驱动下载,我下载的是NVIDIA-Linux-x86_64-418.43.run
3.2安装显卡驱动
下面安装显卡操作建议在命令行进行,安装显卡可能会出现桌面显示不了的问题,倒不如之间在命令行操作,进入命令行操作,进入系统时按:Ctrl+Alt+[ F1 -F6] , F1-F6任选一个,F7是进入图形界面,这里不用;
安装前先更新下配置,重启按Ctrl+Alt+ F1,进入命令行
sudo update-initramfs -u # 更新下配置
reboot # 进入系统前 按Ctrl+Alt+ F1 进入命令行
在命令行找到已经下载好的文件,运行如下命令安装:
sudo bash NVIDIA-Linux-x86_64-418.43.run
显卡查看,出现如下请说明显卡已正常安装,其中 Volatile项是使用率。
nvidia-smi
4. 安装CUDA10
建议前往官网下载选择对应的版本下载并安装。我选择runfile方式安装,文件比较大建议下载到本地再安装,在线安装容易出错,或者改成清华源进行安装,速度快些。
我下载的是: cuda_10.1.105_418.39_linux.run, 如下安装
sudo sh cuda_10.1.105_418.39_linux.run
安装过程中,驱动前面已经安装,就不需要了,其他的默认安装,请注意,X 表示安装,空表示不安装。
添加环境变量
vim ~/.bashrc # 进入 .bashrc文件中,在最后添加如下三行代码
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc # 保存并退出
- 查看CUDA
nvcc -V # 如下图可以看到已经安装成功CUDA10.1版本了。