此记录由参考多位博主的记录总结而来,我记录下,看是否对大家有所帮助
首先禁用nouveau
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
在新建的文件中写入
blacklist nouveau
options nouveau modeset=0
随后使写入文件生效
sudo update-initrams -u
sudo reboot #重启电脑
重启后查看禁用是否成功
lsmod | grep nouveau
如果执行这句话,没有输出则说明已经禁用成功
随后打开ubuntu自带的软件和更新功能,点击附加驱动,随后就会发现有会搜索出很多的驱动(开源、专用、各种版本号的都有);去到英伟达驱动的官网,查看自己的电脑显卡支持的版本号范围,随后在这个范围中回到附加驱动页面,选中比较中间的版本号,并点击应用更改即可。
随后终端重启
sudo reboot
重启后输入
nvidia-smi
如果出现GPU列表则安装成功
如果没有出现,尝试先输入以下语句,随后在点击附加驱动的版本号应用更改
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
以上内容为安装显卡驱动,安装好以后准备安装对应版本的cuda和CUDNN
首先在终端输入
nvidia-smi
可以查看显卡的详细信息
同时上面有会有推荐的cuda的版本,我这里显示是CUDA Version:10.1
随后去官网查看10.1版本对应的cuda安装包,大概下载是2.4G大小;
下载完以后将其保存在单独的文件夹中,并将这个文件夹用终端打开,打开以后输入以下语句:
sudo sh cuda_10.1.243_418.87.00_linux.run
随后在跳出来的安装界面中,在【X】Driver栏按一次enter键,使X消失;
随后选中Install,再按一词enter键即可进行安装;
等待一会儿,安装完以后,输入以下语句添加环境变量
vi ~/.bashrc
再打开的文件中,添加以下语句
export PATH="/usr/local/cuda-10.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH"
使环境变量生效
source ~/.bashrc
在终端输入以下语句,最后出现pass即可视为安装成功
cd /usr/local/cuda-10.1/samples/1-Utilities/deviceQuery
sudo make
./deviceQuery
----------------------------------------------------------------------------------------------------------以下内容为安装对应版本的CUDNN
同样在英伟达的官网下载对应版本的CUDNN,就是下载前需要先注册一个账号
我这里下载的是cudn-10.1-linux-ppc64le-v8.0.4.3.0
随后将下载后的文件夹进行解压,解压后
分别到include和lib对应的文件夹输入以下语句
sudo cp cudnn.h /usr/local/cuda/include/
sudo cp cudnn_version.h /usr/local/cuda/include/
sudo cp libcudnn* /usr/local/cuda/lib64/
随后输入
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/include/cudnn_version.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
以上步骤完成后,即可在终端查看安装的CDNN版本
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2