Ubuntu18.04安装hls4ml+tensorflow-gpu
Ubuntu18.04安装hls4ml+tensorflow-gpu
主要分为两部分,一部分是安装hls4ml
另一个是安装gpu 版本的tensorflow,比较难安装的是tf
1.
https://github.com/fastmachinelearning/hls4ml-tutorial
git clone https://github.com/fastmachinelearning/hls4ml-tutorial.git
2.修改environment.yml文件
- 把tensorflow==2.3.1改为2.4.0
不然报错”AttributeError: module ‘tensorflow.keras.layers’ has no attribute ‘MultiHeadAttention’
根据https://github.com/fastmachinelearning/hls4ml/issues/514和https://github.com/fastmachinelearning/hls4ml-tutorial/pull/27
是由于TensorFlow版本的问题
把name改成环境的名字
删除在这里安装的tensorflow,因为在这里安装同时会安装cuda10的版本,一会儿自己安装
name: hls4ml-new2
channels:- conda-forge
dependencies:- python=3.7- jupyterhub- pydot- graphviz- pip- pip:- jupyter- hls4ml[profiling]==0.6.0- qkeras==0.9.0- scikit-learn- git+https://github.com/thesps/conifer.git
3.
conda env create -f environment.yml
conda activate hls4ml-new2
4.
https://blog.csdn.net/ly869915532/article/details/124627618
然后后面主要按照这个步骤安装tensorflow的gpu版本
https://medium.com/analytics-vidhya/install-tensorflow-gpu-2-4-0-with-cuda-11-0-and-cudnn-8-using-anaconda-8c6472c9653f
安装cuda
conda install cudatoolkit=11.0
5.然后安装tensorflow gpu
pip install tensorflow-gpu==2.4.0
6.
然后在https://developer.nvidia.com/rdp/cudnn-archive 下载这个,需要先登录
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-kBWkUNyc-1656160404541)(https://raw.githubusercontent.com/liuhao-97/typora_pic/main/uPic/image-20220625150947852.png)]
复制这些文件到anaconda3/envs/hls4ml-new2/lib
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-OQNtk9gq-1656160404542)(https://raw.githubusercontent.com/liuhao-97/typora_pic/main/uPic/image-20220625151027584.png)]
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-NoA3jhbN-1656160404543)(https://raw.githubusercontent.com/liuhao-97/typora_pic/main/uPic/image-20220625151105919.png)]
结束
最后在hls4ml-new2环境下输入conda list | grep cuda
只能看到cuda,看不见cudnn
5.
根据https://blog.csdn.net/weixin_45587004/article/details/124408228?spm=1001.2014.3001.5502
修改环境的文件
6.
但是这时候还是用不了GPU 的tensorflow,感觉是env.yml里面TensorFlow版本的问题
https://stackoverflow.com/questions/66389006/how-do-i-install-tensorflow-2-4-on-anaconda
安装conda gpu
conda install tensorflow-gpu==2.4.1
7.
import tensorflow as tf
tf.__version__
查看tensorflow版本,这时候应该是2.4.1
tf.config.list_physical_devices('GPU')
看GPU是否能用
8.
但是不知道为啥好像更慢了?
感觉只能用cpu版本的?
7.
参考
https://medium.com/analytics-vidhya/install-tensorflow-gpu-2-4-0-with-cuda-11-0-and-cudnn-8-using-anaconda-8c6472c9653f
安装pip install tensorflow-gpu==2.4.0
然后
pip show tensorflow-gpu
显示安装位置
几个问题
1. 查看cuda版本
conda list | grep cuda
总而言之就是tensorflow gpu 2.4.0下,默认会安装Cuda 10.2 and cuDnn 7
需要改成正确的[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-cg81MGvY-1656160404543)(https://raw.githubusercontent.com/liuhao-97/typora_pic/main/uPic/image-20220625135202837.png)]
2. 报错 Not creating XLA devices, tf_xla_enable_xla_devices not set
代码里加入
os.environ['TF_XLA_FLAGS'] = '--tf_xla_enable_xla_devices'
3. Tensorflow gpu显存占满
代码里加
# https://blog.csdn.net/weixin_45092662/article/details/117109615
os.environ['CUDA_VISIBLE_DEVICES']="0"
config=tf.compat.v1.ConfigProto()
# config.gpu_options.per_process_gpu_memory_fraction=0.8
config.gpu_options.allow_growth = True
sess=tf.compat.v1.Session(config=config)
4. jupyter搜不到内核
https://blog.csdn.net/wxyczhyza/article/details/120821510
如果在anaconda启动jupyter notebook找不到anaconda的虚拟环境内核,可以尝试以下步骤:
1、在dos窗口输入以下命令激活环境:anaconda activate 【py环境名,比如py37】
2、在虚拟环境py37下安装jupyter notebook,命令:pip install jupyter notebook
3、安装ipykernel,命令:conda install ipykernel
4、将虚拟环境注入jupyter notebook,命令:
python -m ipykernel install --user --name 【环境名称】 --display-name 【在jupyter中显示名字】
如:python -m ipykernel install --user --name py37 --display-name py37(可以和环境名一样,也可以不一样。)
5、运行jupyter notebook,命令:jupyter notebook
Ubuntu18.04安装hls4ml+tensorflow-gpu相关推荐
- Ubuntu18.04安装keras(tensorflow)从无到有实录
今天演示一下在Ubuntu18.04上安装keras (后端tensorflow CPU版本)的过程. 整个过程我们在云上安装. 部署云主机 随便找一家云服务提供商,国内有阿里,腾讯,百度,金山,京东 ...
- Ubuntu18.04安装cuDNN和Tensorflow的正确姿势
文章目录 Ubuntu18.04安装cuDNN和Tensorflow的正确姿势 一.检查NVIDIA驱动是否安装成功 二.检查CUDA是否安装成功 三.安装cuDNN 1. 确定版本 2. 下载安装包 ...
- 笔记本显卡1660Ti,Ubuntu18.04安装Cuda、Cudnn、Anaconda、PyTorch、Tensorflow和Keras
/**********************************************************************/ 注:写到最后不能放图片了,安装PyTorch.Tens ...
- Ubuntu18.04安装CUDA10.1和cuDNN v7.6.5
Ubuntu18.04安装CUDA10.1和cuDNN v7.6.5 前言 深度学习需要使用使用GPU加速,在安装tensorflow之前需要根据选用的版本安装 就我的经验而言,CUDA的安装在Lin ...
- Ubuntu18.04安装python3.7.7和tensorflow-gpu 2.2.0
Ubuntu18.04安装python3.7.7和tensorflow-gpu 2.2.0 前言 安装python3.7.7与环境变量设置 使用pip3安装tensorflow-gpu 2.2.0 测 ...
- 【ubuntu18.04安装】128G u盘安装ubuntu18.04
Ubuntu18.04下安装Nvidia驱动和CUDA10.1+CUDNN_那记忆微凉的博客-CSDN博客_ubuntu18.04安装cuda10.1 Ubuntu 18.0.4安装以及深度学习基本环 ...
- Ubuntu18.04安装CUDA、cudnn、pycharm、tensorflow-gpu
前言 虚拟机里的linux系统似乎不能安装nvidia显卡驱动,在虚拟机Ubuntu系统上跑tensorflow只能使用cpu..看来还得去物理机装双系统或者用服务器了 因为,安装nvidia驱动报错 ...
- Ubuntu18.04安装OpenCV4.3.0和环境配置(支持编译CUDA并安装配置python-opencv)
Ubuntu18.04安装OpenCV4.3.0和环境配置 下载源文件 安装依赖 编译安装 环境配置 动态库配置 配置OpenCV的`PKG-CONFIG`环境 Python-OpenCV环境[如果安 ...
- ubuntu18.04安装显卡驱动,Anaconda,CUDA,pytorch全套流程
记录一次成功的装显卡,以为过程很复杂,其实这边封装的还挺好,正常走流程就没啥问题.我这边也是看各个步骤的博客总结,可以称之为搬运工. 1. 显卡驱动安装 参考的是这篇博客:Ubuntu18.04安装显 ...
最新文章
- 对称加密实现重要日志上报Openresty接口服务
- [转] CSS 颜色名
- INFO:InstallShield InstallScript工程中自定义界面文本输入控件的两个注意事项
- Apache Camel 3.1 –更多骆驼核心优化(第2部分)
- linux下生成coredump文件
- 撸完这些JVM知识点,明天就去面试阿里P6+
- java的简单算法题_[2]十道算法题【Java实现】
- Git的使用的简单介绍
- 进击的程序媛:从 Google 第一位程序媛到硅谷女王进化史
- Codeforces Round #273 (Div. 2) D. Red-Green Towers DP
- java rxtx 串口_Java使用RXTX进行串口SerialPort通讯
- 微信小程序:2022虎年全新头像框制作
- 如何正确地准备KODI媒体文件
- Android多开检测的另一个思路
- 算法探索实录 - 4、递归
- 广义相对论和狭义相对论到底是讲什么的?
- 基于局部均方差的人脸磨皮matlab程序
- 肇庆学院计算机科学与技术专业多少分能进,2020肇庆学院录取分数线是多少-各专业分数线...
- java emoji问号_java过滤emoji字符
- Transformer 五年引用超四万,其中六位作者创立五家创业公司
热门文章
- 澄净是什么意思? - 已回答 - 搜搜问问
- Greenshot — 支持多种插件的截图工具,一键上传网盘
- dl160 g6 linux raid驱动,惠普HP DL160 G6 RAID和Server2003系统安装方法
- 企业数字化转型挑战——知识库管理
- 华为机考108题(c++)(31-40)
- 机器学习9衡量线性回归法的指标,MSE,RMS,MAE
- 流星灯C语言程序,(18)改进led驱动和流星灯程序
- C语言修改同花顺程序吗,炒股软件哪家强?第一篇——同花顺
- 将RGB数据转为图片的MATLAB代码
- 错题本——数据库系统工程师 2009