1.安装显卡驱动
在软件商店的右上角位置点击软件和更新, 找到附加驱动,选择nvidia- driver-450(专有),然后点击应用更改。
重启
运行sudo /usr/bin/nvidia-settings
检查驱动安装情况
运行cat /proc/driver/nvidia/version
查看驱动版本。
2. 安装CUDA
我的显卡是GTX960m,选择cuda9.0,在nvidia官网下载
链接: CUDA.
将第一主文件和下面4个补丁都下载。
- gcc降级
由于cuda9.0不支持gcc7,将其降级为6版本:
sudo apt-get install gcc-6#安装6
sudo apt-get install g++-6
cd /usr/bin
ls -l gcc*
sudo mv gcc gcc.bak #备份
sudo ln -s gcc-6 gcc #重新链接
sudo mv g++ g++.bak#备份
sudo ln -s g++-6 g++#重新链接
运行gcc -v
查看版本为6.5
- 安装cuda
sudo sh cuda_9.0.176_384.81_linux.run
接下来会显示一个文档界面,需要一直按空格读完。
最后会有让你选择的命令