在Ubuntu16上安装Torch7

最新推荐文章于 2025-07-09 15:04:37 发布

Genera1Z

最新推荐文章于 2025-07-09 15:04:37 发布

阅读量1k

点赞数 1

CC 4.0 BY-SA版权

文章标签：深度学习 ubuntu Torch

本文链接：https://blog.youkuaiyun.com/u010175803/article/details/81516211

本文详细介绍了如何在Ubuntu 16.04上安装NVIDIA驱动、CUDA 8.0、cuDNN 5.1以及Torch 7。重点强调了避免使用deb包，推荐使用runfile安装，并提供了每一步的详细操作步骤，包括禁用nouveau驱动、设置环境变量以及验证安装成功的命令。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

虽然时下流行TensorFlow和PyTorch两个深度学习框架，但是仍有些有价值的研究是基于Torch的。

目标组合：Ubuntu 16.04 LTS，NNIDIA GeForce 940MX，CUDA 8.0，cuDNN 5.1，Torch 7。

1 N卡驱动

不要用deb！！！最好使用runfile安装。所有型号的驱动从此处查找并下载：https://www.nvidia.cn/Download/index.aspx?lang=cn。

（1）安装前，先禁用系统自带的N卡驱动nouveau：

sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

在文件中添加以下内容：

blacklist nouveau
options nouveau modeset=0

生成新的kernel initramfs：

sudo update-initramfs -u

（2）重启系统，按Ctrl+Alt+F1进入命令行模式（回到图形模式是Ctrl+Alt+F7），安装下载好的驱动文件：

# 关闭桌面服务
sudo service lightdm stop
sudo chmod u+x ./NVIDIA-Linux-x86_64-390.77.run
# 最后的选项表示只安装驱动文件，不安装OpenGL。不可忽略，否则会导致登录界面死循环
sudo ./NVIDIA-Linux-x86_64-390.77.run --no-opengl-files
# 重启桌面服务
sudo service lightdm start

（3）可以通过以下命令确认驱动是否正确安装：

cat /proc/driver/nvidia/version

2 CUDA

也不要用deb。最好使用runfile安装。可以从此处下载所需版本的CUDA：最新版https://developer.nvidia.com/cuda-downloads；历史版本https://developer.nvidia.com/cuda-toolkit-archive。

（1）按Ctrl+Alt+F1进入命令行模式，执行：

最低0.47元/天解锁文章

200万优质内容无限畅学