ubuntu16.04 配置显卡驱动+cuda8.0+cudnn+pytorch
1. 在线安装显卡驱动
1.ubuntu桌面左上角设置 -> system setting -> software&apdates -> additional drivers -> using NVIDIA binary driver
2.安装完以后 终端输入nvidia-smi
验证驱动是否装好
2. 离线安装cuda
1找到 cuda_8.._linux.run
和 cuda_8..._375.26_linux.run
所在文件夹
2. 键入 ctrl+ shift +F1 进入命令行界面,输入sudo service lightdm stop
关闭图形界面。
3. 在命令行界面输入sudo bash cuda_8_.._375.26_linux.run
注意: 在安装过程中,会询问是否需要装显卡驱动 输入 n (避免重复安装显卡驱动), 其余都选y
4.安装完以后, 输入 ls
可以查看文件列表,通过sudo bash ***.run
在执行安装另一个 cuda文件,一直 enter 直到安装完成
5.启用图像界面 sudo service lightdm start
即可进入
3. 安装cudnn
- 进入cudnn官网,登录后点击 Downloadcudnn 下载cuDNN v7.1.