在安装n卡和cuda的时候,遇到了安装问题NVIDIA都会给出相应的错误提示以及可能的解决方法,很贴心。
写在前面
这两天在知乎看到一篇挺好的安装教程,觉得不错,就保存下来,顺便分享给大家。原文链接
同时,github上还有个cuda,N卡,深度学习环境搭建的安装脚本,大家也可以参开下。醒目的链接
一、安装NVIDIA显卡驱动
环境:
os:ubuntu18.10
GPU:GTX 1080ti
准备工作
ubuntu18.10 自带gdm3 桌面管理,请install lightdm将其替代(gcc,make什么是要安装的,如果没有安装,后边在安装显卡驱动的时候会报错提醒的,don't worry)
开始
屏蔽nouveau(这块儿网上的教程多如牛毛,此处 略)
reboot 重启系统
然后(这块儿网上的教程多如牛毛,此处 略)
最后(这块儿网上的教程多如牛毛,此处 略)
CUDA 的安装
1、下载cuda安装包(因为我安装的NVIDIA显卡是支持cuda10.2的,所以此处下载的是cuda10.2的安装包)
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
2、 安装cuda安装包 sudo sh cuda_10.2.89_440.33.01_linux.run
(cuda10.2的配置很人性化了,都整了个配置页了,其中我取消勾选Driver这个选项,因为我在上边安装了NVIDIA显卡驱动)
CUDA Installer
- [ ] Driver
[ ] 440.33.01
+ [X] CUDA Toolkit 10.2
[X] CUDA Samples 10.2
[X] CUDA Demo Suite 10.2
[X] CUDA Documentation 10.2
Options
Install
3.直接点击Install进行安装,安装成功的页面如下,记得讲cuda路径添加到环境变量里边哦
Please make sure that
- PATH includes /usr/local/cuda-10.2/bin
- LD_LIBRARY_PATH includes /usr/local/cuda-10.2/lib64, or, add /usr/local/cuda-10.2/lib64 to /etc/ld.so.conf and run ldconfig as root
(CUDA的安装到此结束)
CuDNN的安装
见如下链接
https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html