转载自wu496963386 cuda11.1 tensorflow1.15.4进行训练,亲测有效
RTX3080+Ubuntu18.04+cuda11.1+cudnn8.0.4+TensorFlow1.15.4+PyTorch1.7.0环境配置
RTX3080+Ubuntu18.04+cuda11.1+cudnn8.0.4+TensorFlow1.15.4+PyTorch1.7.0环境配置
- 一、驱动及cuda、cudnn的安装
- 二、pytorch的安装
- 1. 源码编译安装
- 2. pip方式安装
- 三、TensorFlow的安装
- ※ nvidia-tensorflow==1.15.4+nv20.10版本的安装
- 1. 安装 TensorFlow wheel的索引
- 2. 安装TensorFlow
- 3. 安装nvidia-TensorFlow对应的依赖包
- 4. TensorFlow测试
- 四、后记
- 参考
有幸公司给购买了RTX3080,可以上手试试,但是发现环境的配置是个头大的问题,且由于项目的关系,至今TensorFlow的版本至今仍是1.x的版本,所以得考虑TensorFlow的具体版本对应,还有pytorch的适配。最终经过各方资料的查找与尝试,终于安装完成,安装好的驱动及各版本如下。
Ubuntu:18.04.5
cuda:11.1
cudnn:8.0.4.30
TensorFlow:1.15.4
pytorch:1.7.0
首先分享安装所有库的网盘链接:
网址:https://pan.baidu.com/s/1j6XSG8oyBX6lzHzka7xr3Q;
提取密码:5tgm
一、驱动及cuda、cudnn的安装
RTX3080采用了新的Ampere架构GA102-200,那么显卡驱动也必然是需要最新的,可以参考cuda与驱动对应的关系:
我们安装的是455.23.04版本,CUDA是11.1版本,cudnn是8.0.4.30版本,具体的安装方法可参考一下网址。
ubuntu16.04系统run方式安装nvidia显卡驱动
Ubuntu16.04下cuda和cudnn的卸载和升级
方法都是大同小异,最后输入nvidia-smi确认驱动是否安装好,nvcc --version确认是否安装好cuda。
二、pytorch的安装
1. 源码编译安装
pytorch的安装相对简单一些,可以选择源码编译安装,也可以选择直接pip的方式安装。由于我的网络实在不好,每次在拉源码的时候都会报错,就放弃了。。依赖库真的太难下载了。具体的源码编译方式可参考下面的方法。
RTX3080/RTX3090驱动安装CUDA11.1+CUDNN8.0.4.30+pytorch源码编译
Pytorch源码编译简明指南
2. pip方式安装
进入pytorch官网,可以看到下载方式,命令:
pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
- 1
也可以进入下载的网站,直接点击下载,然后再通过安装whl的方式进行安装pytorch。
安装完的结果如下图。
安装完再进行测试,版本以及cuda是否可用。
三、TensorFlow的安装
如果没有版本的硬性要求,甚至可以直接通过安装tf-nightly的版本,那就很方便了,直接通过命令:
pip install tf-nightly-gpu
- 1
就可以安装好了,对应版本是2.4+。
※ nvidia-tensorflow==1.15.4+nv20.10版本的安装
经过大量资料的查找,得知TensorFlow官方是已经放弃了1.x的版本更新,那么还有没有还在更新的TensorFlow1.x版本呢?答案是有,终于被我们找到了,参考Accelerating TensorFlow on NVIDIA A100 GPUs。这是nvdia官方对也是安培架构的A100 GPU进行编译的TensorFlow 1.15版本,进行了简单介绍以及安装方法,这对同样是安培架构的3080也是通用的。
具体安装方法简介如下:
1. 安装 TensorFlow wheel的索引
pip install nvidia-pyindex
- 1
2. 安装TensorFlow
pip install nvidia-tensorflow
- 1
大概率是会安装报错,有许多个依赖包,且有些包比较大,如果是不能科学上网的话,很难一次性安装成功。那么优先安装依赖包。
3. 安装nvidia-TensorFlow对应的依赖包
具体依赖包有下:
nvidia-cublas 11.2.1.74
nvidia-cuda-cupti 11.1.69
nvidia-cuda-nvcc 11.1.74
nvidia-cuda-nvrtc 11.1.74
nvidia-cuda-runtime 11.1.74
nvidia-cudnn 8.0.4.30
nvidia-cufft 10.3.0.74
nvidia-curand 10.2.2.74
nvidia-cusolver 11.0.0.74
nvidia-cusparse 11.2.0.275
nvidia-dali-cuda110 0.26.0
nvidia-dali-nvtf-plugin 0.26.0+nv20.10
nvidia-nccl 2.7.8
nvidia-pyindex 1.0.5
nvidia-tensorboard 1.15.0+nv20.10
nvidia-tensorrt 7.2.1.4
tensorflow-estimator 1.15.1
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
且依赖包的版本也需要严格对应(以上版本都是对应TensorFlow1.15.4+nv20.10的),否则会报错无法安装。最后安装nvidia-tensorflow,可以得到版本是1.15.4+nv20.10。安装结果如下图。
考虑到大家都不容易下载,我已经把依赖包下载下来并传到网盘了,大家可以自行下载。依赖包下载地址(提取密码5tgm)。有一些包有安装的先后顺序,大家在尝试之后就知道了,就不详细叙述了(也忘记了0.0)。
4. TensorFlow测试
最后安装完了TensorFlow,尝试一下是否可用。
可以看到TensorFlow已经安装成功,且可以识别到显卡并调用cudnn了。
四、后记
后面尝试训练并推理使用,发现显卡已经是可用的状态了,且性能真的可以比得上rtx2080ti,在训练crnn时发现训练速度甚至能提升20%,推理时与2080ti速度差不多,但价格便宜近一半,确实是性价比很高的卡。
参考
【1】: NVIDIA CUDA Toolkit Release Notes
【2】: RTX3080/RTX3090驱动安装CUDA11.1+CUDNN8.0.4.30+pytorch源码编译
【3】: 手起刀落 RTX 3080 - ResNet50 Benchmark Test
【4】: cuDNN Release 8.x.x
【5】: Accelerating TensorFlow on NVIDIA A100 GPUs
http://www.taodudu.cc/news/show-3359160.html
相关文章:
- Wu-Manber 经典多模式匹配算法
- 5种分布式事务解决方案优缺点对比
- 射线与三角形求交
- 【Flink基础】-- Flink CDC介绍
- Thymeleaf一篇就够了
- 遗传算法详解与改良遗传算法
- 【数论】牛顿迭代法
- Python小白的数学建模课-B6. 新冠疫情 SEIR 改进模型
- Win10家庭版添加组策略gpedit.msc的办法
- 【数学建模】(十):Lingo使用:线性规划+非线性规划+整数规划
- 2020年7月-8月CISP及CISP-PTE线下考试时间
- CISP-PTE2021最新考试经验
- NISP和CISP报名条件
- CISP考试通过率高不高?
- CISP-PTE靶场搭建
- ITSS服务项目经理和服务工程师考试大纲
- CISP-PTE学习笔记综合考试题复现WP
- CISP 老师课堂笔记
- 2022年CISP-PTE/CISP-PTS认证考试要点整理
- CISP-PTE/CISP-PTS认证考试要点整理
- MySql左连接无法显示左表全部数据
- mysql 左连接不生效
- MYSQL左连接查询过滤问题
- MySQL左右连接
- Mysql左连接详解
- MySQL左连接之后引起的排序问题
- mysql左连接多条件,on子句多条件
- 秋招被问mysql左连接和右连接的区别?
- Mysql 左连接 多个字段
- mysql左连接null值丢失、条件不起作用的问题
转载自wu496963386 cuda11.1 tensorflow1.15.4进行训练,亲测有效相关推荐
- IntelliJ IDEA 15在线激活码(破解)--License server 15.0.2本人亲测
IntelliJ IDEA 15 激活码(破解)--License server 15.0.6 本人亲测 http://blog.csdn.net/liyanlei5858/article/detai ...
- 【02月25日】【精彩电影合集】【15部】【亲测】【Lsyq5647发布】
今日电影更新[15部] 1.<绢>07最新多国打造大片DVD中字 2.<寿喜烧西部片>07最新全明星火爆大片DVD中字 3.<美国处男>上亿票房爆笑喜剧DVD中字 ...
- (转载)WebStorm 2018.3.2 激活方式(永久)亲测好用!!!!!!
转载链接:https://blog.csdn.net/qq_31279347/article/details/85225234 前言: 总觉的注册码很费事,过一会儿就失效了,博主觉得还是打补丁的方式一 ...
- RTX3090 与GTX1080Ti Tensorflow1.15训练YOLOv3对比
安装环境: https://blog.csdn.net/wu496963386/article/details/109583045 基本按照上述链接博主方法安装tensorflow1.15.4 本链接 ...
- 1660 super安装tensorflow1.15
1.查看NVIDIA版本信息 其实这里的cuda版本和安装的tensorflow无关!!! 2.查看对应显卡支持的cuda+cudnn 3.经过长期的试错,得出结论.1660 super安装cuda1 ...
- tf.keras.activations.gelu tensorflow1.15.0
鉴于tensorflow1.15.0没有tf.keras.activations.gelu函数,所以需要添加gelu函数的定义. 以下代码为激活函数gelu的定义: def gelu_(X):retu ...
- vegas pro 15解决导入的视频和音频有噪声问题,亲测可行
vegas pro 15解决导入的视频和音频有噪声问题,亲测可行 参考文章: (1)vegas pro 15解决导入的视频和音频有噪声问题,亲测可行 (2)https://www.cnblogs.co ...
- (亲测无坑)Centos7.x使用kubeadm安装K8s集群1.15.0版本
基础环境配置 三台Centos7.x的服务器,主节点 cpu >=2,node节点>=1 注:(上述cpu为最低配置,否则集群安装部署会报错,无法启动,对其他硬件无硬性要求) 以下操作若无 ...
- [转载]Navicat12.1系列破解教程,Navicat12.1.16破解亲测有效!!
[转载: 烟灰哥是传说]Navicat12.1系列破解教程,Navicat12.1.16破解亲测有效!! 2019.2.26月补充: 因CSDN在线更新下载资源比较麻烦,所以,改用baidu网盘免费提 ...
最新文章
- Spring Cloud第七篇:高可用的配置中心
- Servlet与JSP学习笔记(五) JSP核心(下)
- struct和class之间问题(+引发的个人思考,和对共用体思考)
- Mac 上无法使用netstat的-p参数
- 什么是PM2 ---- (零秒重启)
- go项目实战 <微信公众号后台开发>(一、获取token)
- 远程控制软件编写--系列教程
- 【BX学习之软件著作权】 软件著作权的申请
- 机器学习:02 特征工程和决策树回归
- Python concurrent.futures 的 map 函数解释
- 长周期光纤光栅 matlab,长周期光纤光栅传输谱的MATLAB仿真
- 什么是coms coms设置图解
- 3dmax软件常用操作
- 情感分析ASTE三元组提取的BIO标注
- 解决iOS的H5页面卡顿以及键盘回落后留白问题
- 愿我如星君如月,夜夜流光相皎洁
- mysql是后端还是前端_php是前端还是后端
- LaTex 插入表格与图片
- 校园网-校园网wifi-校园网免认证教程
- 上海游趣运营局部全体裁撤 或转型纯研发