Ubuntu16.04下配置CUDA 9.0 + cuDNN 7.1 + tensorflow 1.9环境

本文详细介绍了如何在Ubuntu16.04上安装CUDA 9.0、cuDNN 7.1,并集成到tensorflow 1.9环境。首先确保Nvidia显卡驱动安装正确,然后不选择安装NVIDIA驱动进行CUDA的.run文件安装。接着添加CUDA路径到系统PATH,测试CUDA安装成功。再下载cuDNN进行解压安装,测试通信组件。最后,提供了通过pip在虚拟环境和系统环境中安装tensorflow的方法,以及源码编译的初步介绍。
摘要由CSDN通过智能技术生成

一、确保Nvidia显卡驱动已成功安装

nvidia-smi

 输出大概如下即可,注意看一下版本号

二、安装CUDA9.0:

查看你显卡驱动的版本,然后去Nvidia官网下载相对应版本的CUDA包,选择.run文件,因为deb会把之前显卡驱动的版本给覆盖掉,可能会出很多的问题,,

运行run文件,我的是cuda_9.0.176_384.81_linux.run,后面你替换为你自己的就OK啦

cd到你下载的run文件的目录

chmod +x cuda_9.0.176_384.81_linux.run         #此处为你下载的run文件的名称

./cuda_9.0.176_384.81_linux.run

 

可以慢慢阅读完弹出的那个声明,直接跳过的方式我不告诉你哈哈哈哈

其实是ctrl+c

然后选项中,不选择安装NVIDIA显卡驱动,

然后其他几乎一路默认即可,安装路径可以留意一下,一般默认为/usr/local/cuda-9.0

等待安装完成后,在PATH里添加该CUDA路径,这一步有好多方法的

我选择在/etc里直接添加个全局的,

sudo vim /etc/profile

然后在这个文件末添加如下两句:

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

然后source一下,或者重启,就OK了。

测试是否安装成功:

cd到CUDA样例里,NVIDIA_CUDA-9.0_Samples,然后其实可以全部make一下所

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值