ubuntu18.04安装CUDA+Cudnn+Tensorflow

查看电脑是否支持intel不做赘述;
一定要看好对应的cuda!!!!!
!!!!!!
提前搞好驱动和g++和gcc对应的cuda版本,可以在网上查看一下
!!!!!!
我是安装cuda10.0 + cudnn7.5 + Tensorflow-gpu 1.14(差不多可以跑所有tensorflow1.+版本程序)
一、Cuda下载安装(安装cuda10.0)
直接上连接CUDA10.0下载地址

https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal

盗图一张
在这里插入图片描述
第二个拓展包可以不用添加,如向添加下载运行完cuda接着继续
若cuda安装包保存在:桌面

cd 桌面
sudo sh cuda_10.0.130_410.48_linux.run    # sudo sh + 安装包地址
# sudo sh cuda_10.0.130.1_linux.run   ### 可能会有用

然后直接按 ctrl+c 跳过协议,下面操作:
1.输入accept
Do you accept the previousily read EULA?
accept/decline/quit: accept
2.问是否安装驱动accept接下来的问题
选择:n 千万千万选择 n
剩下的选择y和Enter即可

配置环境:
打开终端输入 gedit ~/.bashrc也可以vim ~/.bashrc(我喜欢用第一个)
在文件最下面添加:记得点击右上角的保存

export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda

注:若安装多版本CUDA需要将其中‘cuda-10.0’中的‘-10.0’删除

然后更新,并检测是否安装成功:

source ~/.bashrc
nvcc -V   # nvcc --version
cat /usr/local/cuda/version.txt

正常出现cuda版本表示安装成功 /(或者)通过下面操作也可以

cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery

返回 Result = PASS 表示安装成功

二、安装cudnn
需要注册一个号用邮箱
下载地址
千万选择对应cuda版本的cudnn,例如我的选择cudnn7.5.0
然后点开对应的cudnn
方法一、可以点击cuDNN Library for Linux即可,下载后解压;
进入解压后的文件夹
在终端运行

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ 
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

方法二、也可以选择(推荐方法一)

cuDNN Runtime Library for Ubuntu18.04 (Deb)

cuDNN Developer Library for Ubuntu18.04 (Deb)

cuDNN Code Samples and User Guide for Ubuntu18.04 (Deb)

三个都要下载(麻烦哦)
然后依次运行这三个包

sudo dpkg -i 地址包

操作完之后可以在终端看看cuDNN的版本:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

cudn10.0 和cudnn7.5.0安装完成

三、安装Tensorflow
建议创建虚拟环境进行管理(以anaconda解释器pycharm为例)

# conda create -n [name] python.version 以我为例(版本号可以换)
conda create -n tf1 python=3.6.5

可能会出错超时,试试加–default-time==100(忘了一个=还是两个了)在pip和install之间

pip install tensorflow-gpu==1.14 -i https://pypi.tuna.tsinghua.edu.cn/simple

测试是否安装成功
在终端输入
python
返回结果true成功了,注:在输入import tensorflow后会出现一堆垃圾,不要在意

>>>import tensorflow
>>>tensorflow.test.is_gpu_available()

最后:如果想更换tensorflow版本可以直接下载安装就会替换原版本

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值