试用阿里云GPU服务器进行深度学习模型训练

蜀国丞相

试用阿里云GPU服务器进行深度学习模型训练

最近在用PyTorch时发现在本地训练模型速度一言难尽，然后发现阿里云可以白嫖gpu服务器，只要没有申请过PAI-DSW资源的新老用户都可以申请5000CU*H的免费额度，三个月内有效。
阿里云免费试用活动页面
一、申请试用并创建实例

• 点击试用，完成注册、实名、领取产品，然后前往控制台
  
• 创建工作空间并授权
  
• 授权完成，前往默认工作空间
  
• 创建交互式建模（DSW）实例
  找到交互式建模（DSW）然后点击创建实例
  
  
  
  
  输入实例名称（随便写）
  
  
  接下来选择GPU规格，选 A10 或者 V100 只有这两个是支持资源包抵扣的（资源包在我们申请试用的时候已经自动领取了），其他的不支持，这里注意！我这里选的是 A10显卡，每小时消耗6.991计算时，性能已经够用了。V100显卡的性能更好，相应的资源包抵扣使用时间也会短一些，大家可以根据自己需求选择。
  
  
  然后选择镜像，点击下一步创建实例即可
  
  
  

二、部署代码到实例进行训练

• 启动并打开实例
  
  
  打开后可以看到导航栏中有三项
  
  
  
• 上传训练模型的py文件
  点击这个按钮上传
  
  
  上传后可以在侧边栏看到
  
  
  也可以打开terminal查看，同样可以看到
  
  
  
• 开始训练
  跟在本地是一样的，依次执行generate.py、loader.py、train.py，我这里选择在terminal用命令执行，大家也可以选择在集成的vscode中执行
  1. # 在terminal依次输入下列命令执行
  2. python generate.py # 可能会提示ModuleNotFoundError: No module named 'captcha'，这是我们的环境中没有captcha这个包，用pip安装即可，pip install captcha
  3. python loader.py
  4. python train.py

  复制代码
• 下载训练的pth文件到本地
  训练完成后会有一个pth文件，这是PyTorch训练的模型文件，我们可以下载到本地使用
  
  
  

三、总结

以上就是在阿里云人工智能平台PAI使用交互式建模（DSW）训练深度学习模型的过程，欢迎大家在评论区留言一起交流学习，有需要模型训练过程以及代码文件的可以看我上一篇blog

来源:https://www.cnblogs.com/gggod/p/18140592
免责声明：由于采集信息均来自互联网，如果侵犯了您的权益，请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容，谢谢合作！