博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
使用Google-Colab训练PyTorch神经网络
阅读量:6593 次
发布时间:2019-06-24

本文共 1237 字,大约阅读时间需要 4 分钟。

Colaboratory 是免费的 Jupyter 笔记本环境,不需要进行任何设置就可以使用,并且完全在云端运行。关键是还有免费的GPU可以使用!用Colab训练PyTorch神经网络步骤如下:

1:新建Colab文件

Colab是在Google硬盘上面运行的,所以,需要到Google硬盘上面新建Colaboratory文件,并进行关联,文件是以ipynb结尾的Jupyter笔记本。下面有一些Jupyter笔记本的使用技巧,可以帮你更好的使用这个环境:

直接运行python代码

import randomx = 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 drivedrive.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

你可能感兴趣的文章
【MongoDB】Capped固定集合
查看>>
vmnet0 网桥链接报错
查看>>
Spring Boot 最佳实践(二)集成Jsp与生产环境部署
查看>>
找回消失的内存
查看>>
普世智慧 - 跨学科的知识和思维
查看>>
Linux内核高性能优化【生产环境实例】
查看>>
二、搭建Apache虚拟主机
查看>>
罗森伯格参加2012中国国际建筑智能化峰会北京站
查看>>
ospf(专题二)路由再发布
查看>>
HDFS相关
查看>>
我的友情链接
查看>>
windows server 2016 活动目录(一)
查看>>
linux Less
查看>>
我的友情链接
查看>>
Flex 图表组件使用示例
查看>>
嵌入式arm linux环境中gdb+gdbserver调试
查看>>
Mysql数据库备份和还原
查看>>
我的友情链接
查看>>
Cisco ACL访问控制列表实现网络单向访问
查看>>
【翻译】Ext JS 6有什么新东西?
查看>>