您的位置:首页 > 其它

深度工具安装(NVIDIA+cuda+cudnn+tensorflow)

2017-12-20 20:42 686 查看
显卡驱动安装

查询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  

添加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 

 检查cuda版本

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

终端执行
[python] view plain copy 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消失了。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐