Ubuntu 16 安装 torch7 、显卡驱动、CUDA、CUDNN

此步骤安装的是 pytorch 的前身 torch7

硬件环境

Ubuntu 16.04

2080ti * 1

apt 安装 git

（建议执行）

apt

换国内源，极大提升

apt install

速度：

ubuntu、conda、pip 换北外源（北京外国语大学开源软件镜像站）

sudo apt install git

（建议执行）

git

挂代理或镜像站：

git clone一个github上的仓库，太慢，经常连接失败，但是github官网流畅访问，为什么？

安装 torch

注：此步骤最好在未安装驱动、

CUDA

、

CUDNN

时进行，否则 bug、冲突多多多多多多多。

重要：首先将

export TORCH_CUDA_ARCH_LIST="6.0"

写入

~/.bashrc

末尾，此环境变量在于限制硬件算力至

CUDA

版本对应上限（

CUDA8.0

对应

6.0

）

torch

安装命令参考官方文档即可（不用

sudo

）：

git clone https://github.com/torch/distro.git ~/torch --recursive
cd ~/torch; bash install-deps;
./install.sh
source ~/.bashrc
           

注：卸载

torch

只要

rm -rf ~/torch

即可，但同时也建议把

.bashrc

末尾的

torch

环境变量删除。

如果已安装驱动、

CUDA

、

CUDNN

，建议卸载后再安装

torch

，否则有硬件算力、各种版本冲突。

安装（卸载）显卡驱动

推荐在 安装（卸载） CUDA

同样推荐 下载 后安装；

此例中使用的是

cuda_8.0.44_linux.run

；

安装命令：

sudo sh cuda_8.0.44_linux.run

；

安装后在

~/.bashrc

里添加以下两行（配置环境变量）：

export  PATH=/usr/local/cuda-8.0/bin:$PATH
export  LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64$LD_LIBRARY_PATH
           

保存退出后执行

source ~/.bashrc

激活环境变量。

卸载命令：

cd /usr/local/cuda-8.0/bin/
sudo ./uninstall_cuda_8.0.pl
           

安装（卸载）CUDNN

同样推荐 下载（需简单注册登录）后安装；

此例中使用的是

libcudnn7_7.1.3.16-1+cuda8.0_amd64.deb

；

安装命令：

sudo dpkg -i libcudnn7_7.1.3.16-1+cuda8.0_amd64.deb
sudo cp /usr/local/cuda-8.0/lib64/libcudart.so.8.0 /usr/local/lib/libcudart.so.8.0 && sudo ldconfig
           

卸载命令：

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
           

安装 torch 库

# 不用 sudo 
luarocks install cudnn
luarocks install cutorch
luarocks install tds
luarocks install gnuplot
luarocks install xlua
...