前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >使用Pytorch训练手语识别模型

使用Pytorch训练手语识别模型

原创
作者头像
用户9752602
发布2022-05-16 16:27:24
9030
发布2022-05-16 16:27:24
举报

本文将介绍 腾讯云 GPU服务器 GPU计算型GN8 上进行的Pytorch模型训练。

一、实例环境

  • 操作系统:ubuntu 18 (ubuntu or windows 2选1)
  • GPU:Tesla P40(显存24G,超好用) 一块
  • CUDA: 10.2 其他配置:miniconda、scp工具...

二、环境配置

1.Nvidia驱动安装

step1:禁用nouveau

创建blacklist-nouveau.conf

代码语言:txt
复制
sudo vim /etc/modprobe.d/blacklist-nouveau.conf  

在文件中写入

代码语言:txt
复制
blacklist nouveau
代码语言:txt
复制
options nouveau modeset=0

重新配置内核文件

代码语言:txt
复制
sudo update-initramfs -u

最后重启

代码语言:txt
复制
sudo reboot

step2:安装指定版本的驱动

查看可用的驱动

代码语言:txt
复制
ubuntu-drivers devices

安装驱动(以 nvidia-driver-470 为例,具体安装版本按个人要求)

代码语言:txt
复制
sudo apt install nvidia-driver-470

安装完毕之后执行

代码语言:txt
复制
nvidia-smi

查看是否安装成功

2. CUDA & cudnn安装

cuda

使用 wget 可以直接下载,具体下载链接见cuda官网:https://developer.nvidia.com/cuda-toolkit-archive

cudnn

下载稍微麻烦一点,由于我使用的服务器只有命令行,因此先在个人电脑上下载cuda对应的cudnn版本,通过scp上传到服务器上。

方法可能不是最佳解决方法,如有更佳方法欢迎指点

需要注意几点:

①cuda和cudnn安装的 版本对应,下载cudnn的时候一定要注意;

②安装完成之后检查一下:

cuda检查

代码语言:txt
复制
nvcc -V

cudnn检查(以安装地址 /usr/local/cuda 为例)

代码语言:txt
复制
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 

3. miniconda安装

miniconda相较于 conda 而言体量小,功能差不多。

miniconda网址:https://docs.conda.io/en/latest/miniconda.html

复制需要的版本下载链接,使用 wget 可以下载软件包

三、模型训练

这一部分主要配置miniconda环境上传代码运行代码就好了。

为了更直观的修改文件和查看结果,我使用了 MobaXterm 软件登陆服务器。

好处:能点击文件进行修改,上传下载都比较方便,一般不怎么会突然终端。

last but not least

致谢

非常感谢腾讯云平台提供的 free 服务器一个月使用体验,使用体验用两个字总结:畅快。同时也感谢各位前辈的指点和帮助,在使用服务器过程中,我遇到了一些问题,在官方前辈的解答和大佬群友们的帮助下成功解决,很喜欢这样的工作学习氛围,共同进步、携手同行。

一个月的使用时间,让我解决了一大部分论文所需的实验,衷心感谢提供支持的腾讯云平台以及幕后付出的所有前辈。

目前达到的最佳实验结果(WER):26.1%

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 一、实例环境
  • 二、环境配置
    • 1.Nvidia驱动安装
      • step1:禁用nouveau
      • step2:安装指定版本的驱动
    • 2. CUDA & cudnn安装
      • cuda
      • cudnn
      • 需要注意几点:
    • 3. miniconda安装
    • 三、模型训练
    • last but not least
    相关产品与服务
    GPU 云服务器
    GPU 云服务器(Cloud?GPU?Service,GPU)是提供 GPU 算力的弹性计算服务,具有超强的并行计算能力,作为 IaaS 层的尖兵利器,服务于深度学习训练、科学计算、图形图像处理、视频编解码等场景。腾讯云随时提供触手可得的算力,有效缓解您的计算压力,提升业务效率与竞争力。
    领券
    问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档


    http://www.vxiaotou.com