显卡驱动安装
查询nvidia驱动可用版本,运行sudo apt-cache search nvidia-* 查询相应版本
根据得到的结果,安装相应的版本。
2、官网下载显卡驱动包。
lspci | grep -i nvidia (找自己机子的显卡型号)
- sudo add-apt-repository ppa:graphics-drivers/ppa
- sudo apt-get update
- sudo apt-get install nvidia-(版本号)
3、重启
- sudo shutdown -r now
4、nvidia检查
- cat /proc/driver/nvidia/version
CUDA安装
下载cuda最新版本:
- sudo dpkg -i (文件名).deb
- sudo apt-get update
- sudo apt-get install cuda
- echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
- echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
- source ~/.bashrc
- nvcc -V
sudo shutdown -r now
cudnn安装,选择下载的文件
- tar xvf cudnn*.tgz
- cd cuda
- sudo cp */*.h /usr/local/cuda/include/
- sudo cp */libcudnn* /usr/local/cuda/lib64/
- sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
- nvidia-smi
Tensorflow安装(清华大学镜像)
sudo apt-get install python-pip python-dev
sudo pip install --upgrade pip
sudo pip install \ -i https://pypi.tuna.tsinghua.edu.cn/simple/ \ https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.2.0-cp27-none-linux_x86_64.whl在sudo ldconfig时遇到
usr/local/cuda-8.0/lib64/libcudnn.so.5 不是符号连接的问题,解决办法也很简单,重新建立链接并删除原链接
首先找到usr/local/cuda-8.0/lib64/目录,搜索 libcudnn 然后发现
两个文件
libcudnn.so.5 和libcudnn.so.5.0.5 理论上只有一个libcudnn.so.5.0.5
终端执行
- ln -sf /usr/local/cuda-8.0/lib64/libcudnn.so.5.0.5 /usr/local/cuda-8.0/lib64/libcudnn.so.5
再sudo ldconfig时就可以了,这时候会发现usr/local/cuda-8.0/lib64/目录下只有
libcudnn.so.5.0.5 文件了,libcudnn.so.5消失了。