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.3 for CUDA 8.0中的对应16.04的3个Deb文件.(也可以通过
.tzr文件安装,但我更偏向用deb文件,比较方

本文详细介绍了在Ubuntu16.04上配置显卡驱动、CUDA8.0、CUDNN以及PyTorch的步骤,包括在线和离线安装方法,以及环境变量配置,最后进行了PyTorch的测试,确保安装成功。
最低0.47元/天 解锁文章
3656

被折叠的 条评论
为什么被折叠?



