nvidia
文章平均质量分 68
TFATS
这个作者很懒,什么都没留下…
展开
-
基于TensorFlow1使用Horovod实现BERT在单节点上的多GPU卡训练
一,下载从github中下载tensorflow1版本的bert代码和预训练模型1,下载bert代码地址:google-research/bert2,下载预训练模型,此处可选择bert-base中文版本,如下图:3,如官网所示创建执行脚本-方便传参,如下图所示:二,修改创建执行脚本根据机器的GPU个数,Horovod运行多卡训练gpu_num=$(nvidia-smi --query-gpu=name --format=csv,noheader | wc -l)horovodrun原创 2021-09-01 15:48:49 · 892 阅读 · 1 评论 -
Linux中使用Docker快速搭建Tensorflow-Gpu开发环境
本文将介绍:如何找到所需要的Tensorflow-GPU镜像在Linux终端中拉取镜像、查看本地镜像使用Docker构建Tensorflow-Gpu环境配置jupyter外部访问映射检查是否是GPU环境一,下载Tensorflow镜像1,找到所需要的Tensorflow镜像百度搜索nvidia ngc进入NVIDIA GPU 加速容器,找到所需版本的Tensorflow镜像,并copy 其对应的 pull tag。2,在Linux终端中拉取镜像docker pull nvcr.io/原创 2021-08-26 11:21:00 · 2048 阅读 · 0 评论 -
Linux-ubuntu系统查看显卡型号、显卡信息详解、显卡天梯图
一,查看显卡型号1,查看显卡型号命令lspci | grep -i vga#----output------01:00.0 VGA compatible controller: NVIDIA Corporation Device 1f06 (rev a1)可以看到,查看到的显卡型号名称是一个十六进制数字代码。接下来将得到的十六进制数字代码,带入网址中查询具体显卡型号。2,以下网址输入数字代码查询显卡型号The PCI ID Repository可以按照十六进制数字代码找到相应的显卡型号原创 2020-10-19 15:22:48 · 46857 阅读 · 2 评论