上验证cudnn是否安装成功_ubuntu18.04 安装cuda、cudnn、tensorflow和pytorch其实很简单...
目录
一. 安装NVIDIA驱动
1. 首先,查看NVIDIA驱动是否已安装
2.安装NVIDIA驱动
二.安装cuda
1. 查看NVIDIA版本对应的cuda版本
2. 选择CUDA版本(上限原则)
3. 安装cuda
4. 安装cndnn
三.安装tensorflow、pytorch深度学习框架
1. 安装tensorflow-gpu
2. 安装pytorch
四.参考
一. 安装NVIDIA驱动
1. 首先,查看NVIDIA驱动是否已安装
在终端输入以下命令,如果显示以下结果,说明NVIDIA驱动已经正常安装,否则,需要重新安装;
nvidia-smi
2.安装NVIDIA驱动
方法一:使用Ubuntu自带的镜像源安装
第一步:先在软件中搜索“软件和更新”,然后选择一个可用的镜像源,我选择的是清华的镜像源。我的选择如下:
第二步:选择源之后,还是在软件更新中,选择附加驱动,然后选择合适的驱动,应用更改就好,我本机选择的是nvidia-driver-455的驱动。
第三步:完成安装后,在终端输入 nvidia-smi,如果可以显示以下结果,恭喜你,安装NVIDIA驱动成功啦。
方法二:下载NVIDIA官方驱动
第一步: 进入NVIDIA官方下载与之匹配的驱动: https://www.nvidia.cn/geforce/drivers/ ,下图是我本机驱动匹配的信息。点击“开始搜索”,进入NVIDIA驱动下载界面。
第二步:点击获取下载,下载驱动,可以选择第一个最新版本的驱动进行下载安装。我本机选择下载的版本是455.38。
第三步:进入下载好的NVIDIA驱动的路径,开始安装NVIDIA驱动,执行以下命令进行安装。
sudo sh NVIDIA-Linux-x86_64-455.38.run
第四步:验证是否安装成功
完成安装后,在终端输入 nvidia-smi,如果可以显示以下结果,恭喜你,安装NVIDIA驱动成功。
二.安装cuda
1. 查看NVIDIA版本对应的cuda版本
进入该网址查看:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html ,如下图所示;
2. 选择CUDA版本(上限原则)
如上图所示,Ubuntu的显卡驱动版本为455.38,可以支持安装CUDA11.1.1update 1。有一个重要的原则:上限原则,即高版本可以兼容低版本(我本机的NVIDIA驱动支持安装CUDA11.1.1 update 1,所以我原则上可以安装CUDA11.1.1 update 1,及以下的任何一种cuda版本,也就是说,我不仅可以装CUDA11.1.1 update 1,还可以装CUDA10.0,CUDA9.2等,但不能安装CUDA 11.1.1 update 1以上的版本,这是由于受硬件的限制。这可以形象的理解为:一个可以挑100公斤重担的人,正常情况下他可以挑100公斤及100公斤以下的任何重量的物体。但不能挑超过100公斤物体。)
3. 安装cuda
第一步:进入cuda下载页面:https://developer.nvidia.com/CUDA-toolkit-archive
第二步:安装cuda10.0:下载deb文件:https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=deblocal
第三步:文件大小1.6GB,下载速度还是挺快的,下载完成后,进入下载目录,然后在终端输入以下命令:
sudo dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-0-local-10.0.130-410.48_1.0-1/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
第四步:如果发现本机已自动升级最新版本了,运行以下命令
sudo apt-get install cuda-10.0
第五步:CUDA安装完成后,添加环境变量
gedit ~/.bashrc
第六步:在文件最后添加
export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64
${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
第七步:最后,添加的环境变量生效,执行以下命令
source ~/.bashrc
第八步:查看cuda是否安装成功!
方法一:在终端输入
nvcc -V
如果显示以下结果,则表明安装成功:
方法二:去安装目录下查看或在终端输入
cat /usr/local/cuda/version.txt
如果显示以下结果,则表明安装成功:
4. 安装cndnn
第一步:去官网下载https://developer.nvidia.com/rdp/cudnn-archive,下载与cuda版本一致的cudnn版本。我下载的版本是cuDNN v7.6.5版本。
第二步:下载完成后,对文件进行解压,有一个cuda的文件夹,包括include 和 lib64,把里面的文件复制到/usr/local/cuda/里面相应的目录里:
sudo cp 解压地址/cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp 解压地址/cuda/lib64/libcudnn* /usr/local/cuda/lib64/然后给文件加读权限
第三步:然后给文件加读权限
sudo chmod a+x /usr/local/cuda/include/cudnn.h
sudo chmod a+x /usr/local/cuda/lib64/libcudnn*
第四步:更新网络链接
cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.7.6.5 # 自己查看.so的版本
sudo ln -sf libcudnn.so.7.6.5. libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig
第五步:查看cudnn是否安装好:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
第六步:显示如下结果,恭喜,说明cudnn安装好啦!
三.安装tensorflow、pytorch深度学习框架
1. 安装tensorflow-gpu
第一步:查看tensorflow-gpu对应的cuda版本:https://tensorflow.google.cn/install/source_windows
由于我本机NVIDIA驱动支持安装的是cuda11.1.1 update1,(我本机装的是cuda10.0),所以我可以装tensorflow-gpu的cuda11.1.1 update1及以下的任何一个cuda版本。
第二步:安装tensorflow-gpu=2.2.0,进入tensorflow对应的虚拟环境(前提是已安装好了anaconda),在终端输入以下命令:
#使用conda安装tensorflow-gpu,它会自动匹配要安装的cuda版本:cuda10.1,如下图所示
conda install tensorflow-gpu==2.2.0
第三步:验证tensorflow-gpu是否安装成功
进入python编译环境,输入以下代码,如果输出结果是本机的GPU的信息和结果True(我的输出结果是: device (0): GeForce GTX 1080, Compute Capability 6.1 True),表示GPU可用:
python
import tensorflow as tf
print(tf.test.is_gpu_available())
tf.test.is_gpu_available()
2. 安装pytorch
第一步:匹配要安装的pytorch版本:https://pytorch.org/get-started/locally/
第二步:在终端输入以下命令:
conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
第三步:验证pytorch是否安装成功,进入python编译环境,输入以下代码,如果输出结果是cuda,表示GPU可用,如果输出cpu,表示GPU不可以用:
import torch as t
device = t.device('cuda' if t.cuda.is_available() else 'cpu')
print(device)
四.参考
https://blog.csdn.net/momaojia/article/details/104435468#commentBox
上验证cudnn是否安装成功_ubuntu18.04 安装cuda、cudnn、tensorflow和pytorch其实很简单...相关推荐
- ubuntu查看cudnn是否安装成功_深度学习之目标检测系列(0) -ubuntu18.04+RTX2080Ti+cuda+cudnn安装...
ubuntu18.04使用cudnn的过程中会遇见很多问题,而且搜索很多文章发现比较杂乱,我这里总结一下,方便小伙伴少走几步坑. 我的服务器硬件配置信息 (2019-01购买) CPU: 英特尔(in ...
- 从无到有 Ubuntu16.04 18.04 20.04安装+Todesk+Chrome+NVIDIA驱动+CUDA+Cudnn+Anaconda3+Pycharm 超详细教程+踩坑问题
从无到有 Ubuntu16.04 18.04 20.04安装+Todesk+Chrome+NVIDIA驱动+CUDA+Cudnn+Anaconda3+Pycharm 超详细教程+踩坑问题(有部分图片忘 ...
- Ubuntu18.04安装g2o,及问题解决_HyperZhu的博客-CSDN博客_ubuntu18.04安装g2o
Ubuntu18.04安装g2o,及问题解决_HyperZhu的博客-CSDN博客_ubuntu18.04安装g2o sudo apt-get install cmake libeigen3-dev ...
- python3.6.5安装教程-Ubuntu16.04安装python3.6.5步骤详解
下载python3.6.5安装包 1. 上传安装包.打开终端,利用命令cd 进入文件所在文件夹里 python@ubuntu:~/workspace$pwd /home/python/workspac ...
- ubuntu安装python3.6_Ubuntu16.04安装python3.6.5步骤详解
下载python3.6.5安装包 1. 上传安装包.打开终端,利用命令cd 进入文件所在文件夹里 python@ubuntu:~/workspace$pwd /home/python/workspac ...
- python3.6.5下载安装教程_Ubuntu16.04安装python3.6.5步骤详解
下载python3.6.5安装包 1. 上传安装包.打开终端,利用命令cd 进入文件所在文件夹里 python@ubuntu:~/workspace$pwd /home/python/workspac ...
- python3.6.5安装步骤-Ubuntu16.04安装python3.6.5步骤详解
下载python3.6.5安装包 1. 上传安装包.打开终端,利用命令cd 进入文件所在文件夹里 python@ubuntu:~/workspace$pwd /home/python/workspac ...
- Ubuntu安装nvidia显卡驱动和CUDA CUDNN autoinstall
Ubuntu安装nvidia显卡驱动和CUDA CUDNN autoinstall 首先安装nvidia显卡驱动 打开终端输入: ubuntu-drivers devices 终端界面还显示可以安装的 ...
- TVM学习(一)安装TVM Ubuntu16.04安装TVM0.8+clang-llvm12.0
TVM安装 ****下载TVM安装包地址(0.8版本缺少文件vta-hw) Download 一.从github上下载TVM git clone --recursive https://github. ...
- mysql安装后第一次操作_MySQL数据库之mysql 安装成功以及第一次安装成功初始化密码操作...
本文主要向大家介绍了MySQL数据库之mysql 安装成功以及第一次安装成功初始化密码操作 ,通过具体的内容向大家展现,希望对大家学习MySQL数据库有所帮助. 一 把文件解压到一个目录下 技术分享图 ...
最新文章
- 凡人学习Linux之路
- c#异步socket
- mysql plugin 调用_MySQL插件接口的调用方式
- 【JavaSE02】Java基本语法-概念
- 那些2019年会爆发的泛娱乐黑科技风口——网易MCtalk泛娱乐创新峰会揭秘
- 延时队列实现的几种姿势
- 自动控制原理第二版王建辉_王建辉自动控制原理配套题库名校考研真题课后答案资料课后习题章节题库模拟试题...
- 【POJ - 3494】Largest Submatrix of All 1’s(加一点思维后化成 单调栈)
- jQuery使用最广泛的javascript函数库
- css取消聚焦边框[Chrome,Safari]
- IS-IS详解(十七)——IS-IS 区域迁移与路由汇总
- python语法简洁清晰、特色之一是强制用作为语句缩进_问道python之基础篇【一】 认识python...
- html表单代码有哪些,HTML常用代码有哪些
- html5 打开支付宝app,支付宝H5唤醒APP
- 「CSS」知识点笔记:transition
- java调用百度地图的不同显示方式
- 算是我看源码时的一个小技巧吧~
- 《当我谈跑步时我谈些什么》书摘
- 正确设定idea的java的jre环境和jdk
- matlab 求留数,用matlab求留数
热门文章
- idea 新建java类 乱码,Intellij IDEA 2019 最新乱码问题及解决必杀技(必看篇)
- 智联招聘基于Apache Pulsar打造企业级事件中心
- J 位操作练习 (Java)
- 一次性去掉Word 2013文档中所有文字下波浪线的方法
- 视频直播技术(三):低延时直播经验总结
- JQuery iframe页面通过parent方法操作父页面中的元素与方法(实例讲解)
- linux解压缩及源码安装
- 黑客攻击成网络安全大患 危害长久
- Spring的bean管理(注解创建对象)
- java age_这些Java9 超牛的新特性,你竟然还没用过?