使用Google-Colab训练PyTorch神经网络
Colaboratory 是免费的 Jupyter 笔记本环境,不需要进行任何设置就可以使用,并且完全在云端运行。关键是还有免费的GPU可以使用!用Colab训练PyTorch神经网络步骤如下:
1:新建Colab文件
Colab是在Google硬盘上面运行的,所以,需要到Google硬盘上面新建Colaboratory文件,并进行关联,文件是以ipynb结尾的Jupyter笔记本。下面有一些Jupyter笔记本的使用技巧,可以帮你更好的使用这个环境:
直接运行python代码
import random
x = random.random()
print(x)
以 !开头的命令可以执行操作系统的指令
! pwd
! ls
! cd .. # 切换了外面的工作目录,但是不切换当前程序的工作目录
以 %开头的命令表示魔法指令
%matplotlib inline # 内嵌画图
%cd .. # 切换当前程序的工作目录,作用和os.chdir(path) 相同
2:安装pytorch
安装方法一:
! pip3 install torch torchvision # 省事
安装方法二:
! uname -a # 查看系统 x86_64 GNU/Linux
! python --version # 查看python版本 Python 3.6.7
! nvcc --version # 查看cuda版本 release 9.2, V9.2.148
在pytoch下载页面找到合适的版本,进行安装
! pip install https://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
! pip install torchvision
3:挂载Google硬盘
在运行代码的时候需要从Google硬盘读取数据,需要将结果保存到硬盘当中,这个时候需要进行挂载,期间需要输入一个验证码。
from google.colab import drive
drive.mount('/content/drive/') # 将google硬盘挂载在/comtent/drive/目录上面
4:设置GPU
只需要在笔记本中的:修改->笔记本设置 当中就可以找到GPU选项。
5:编写代码,进行训练
直接编写代码进行训练,要注意当前的工作目录,可以使用%cd path
进行切换
目前colab不稳定,连接经常断掉,所以当训练的次数过多、时间过长的时候注意保存实验结果。
如何执行python文件?
当然你也可以不用Jupyter笔记本,使用以.py结尾的python文件,这个时候只需要执行下面的代码。我没有测试在这种情况下是否能够时候用GPU。
! python test.py
转载于:https://www.cnblogs.com/jiaxin359/p/10234946.html
使用Google-Colab训练PyTorch神经网络相关推荐
- yolov4用Google Colab训练自己的数据集
yolov4训练数据 简介 运用google Colab训练自己的Yolov4数据集 google colab的使用 项目使用流程 1.设置GPU可用 2.clone项目构建Darknet git c ...
- 学生党 白嫖之GPU Google colab 训练深度学习模型
白嫖GPU之Google colab 前言:本人是一名穷且孤单的大学秃头生,跟着老师学习的是计算机视觉领域,学习目标检测到目标跟踪已经有一年的时间了,大多数都是在做论文阅读,很少有机会去实战(因为穷, ...
- 神经网络学习小记录69——Pytorch 使用Google Colab进行深度学习
神经网络学习小记录69--Pytorch 使用Google Colab进行深度学习 注意事项 学习前言 什么是Google Colab 相关链接 利用Colab进行训练 一.数据集与预训练权重的上传 ...
- 独家 | 教你使用Keras on Google Colab(免费GPU)微调深度神经网络
作者:LONG ANG 翻译:闫晓雨 校对:丁楠雅 本文约2300字,建议阅读7分钟. 本文将指导您如何使用Google上的Keras微调VGG-16网络. 简介 在CPU上训练深度神经网络很困难.本 ...
- 【写博客常用】Colab Pytorch神经网络基本构建
[写博客常用]Colab & Pytorch调用基本模型 1 使用Colab 2 连接GPU 3 tensor 4 计算梯度 5 读取data,使用torch.utils.data里面的Dat ...
- 送你9个快速使用Pytorch训练解决神经网络的技巧(附代码)
来源:读芯术 本文约4800字,建议阅读10分钟. 本文为大家介绍9个使用Pytorch训练解决神经网络的技巧 图片来源:unsplash.com/@dulgier 事实上,你的模型可能还停留在石器时 ...
- 利用colab保存模型_在Google Colab上训练您的机器学习模型中的“后门”
利用colab保存模型 Note: This post is for educational purposes only. 注意:此职位仅用于教育目的. In this post, I would f ...
- PyTorch 从零训练卷积神经网络(Convent)
本文主要介绍从从零训练卷积神经网络(Convent).使用PyTorch创建各自的convent或神经网络样本. 原文地址:PyTorch 从零训练卷积神经网络(Convent)
- Google Colab 使用教程(1)
Google Colab免费GPU使用教程(一) 免费用户只能看到二级目录,子目录看不到了 !cd /content/YOLOv5_with_BiFPN !pwd !cd YOLOv5_with_Bi ...
最新文章
- 第五周项目一-三角形类雏形(1)
- AOC的显示器真的很烂
- 周志华 机器学习 笔记
- 第一次运行Spring Boot有感
- netflix zuul 1.x与zuul2.x之比较
- OpenResty安装--增强版的nginx
- .NET Core/.NET之Stream简介
- [渝粤教育] 中国地质大学 大学英语(1) 复习题
- assertionerror python_Python成为专业人士笔记–内置模块Modules和函数Functions
- java分割例子,Java 分割字符串详解及实例代码
- vscode中打开pdf文件_提取pdf文件中的文字
- leetcode 88 Merge Sorted Array
- java 32 64 性能,Java 64位的性能是否优于32位版本?
- 【设计模式】适配器模式(Adapter Pattern)
- Request 对象
- 尼康d850相机参数测试软件,尼康D850 这可能是你唯一需要的单反相机
- 屏幕录像软件Community Clips Recorder简介及其使用技巧(郝宪玮)
- 从零搭建个人资讯系统1
- 5.Lucene 基本原理(六)字典实现原理 FST(Finite State Transducer)
- request body is missing