使用Google colab训练神经网络

1、
前提条件:可以科学上网,拥有谷歌账号

2、
创建Google drive

3、
进入到google drive中,点击New,新建一个Folder,命名为“craters”(这里自己取名字)

可以看到

在这里插入图片描述

4、
右键,点击connect more apps,然后搜索colaboratory,并且添加

在这里插入图片描述
在这里插入图片描述

5、 然后右键,more->google colaboratory,建立了一个后缀名为.ipynb的文件,在,点击右上角修改,点击笔记本设置,在硬件加速器里面设置为GPU加速。

在这里插入图片描述

6、注意:输入完代码,按下Shift(先按)+Enter会执行代码,同时跳到下一行。

输入from google.colab import drive

drive.mount(’/content/gdrive’)

目的是挂载Google Drive。点击蓝色URL,按步骤操作,会返回一串秘钥。复制然后粘贴到如下界面输入框内然后点击键盘上Enter(图中已经输入完毕)。
在这里插入图片描述

7、输入命令: !(空格)nvidia-smi查看GPU型号

这里,是Tesla P100,如果我没记错的话算力应该是7.5

在这里插入图片描述

8、安装tensorflow2.1.0版本(下面命令语句就省略不打了)

在这里插入图片描述

注意:左上角的圈圈转的时候不要执行下一句代码,否则容易出错

9、如果要使用keras,由于keras的很多代码是建立在tensorflow1.x版本上的,由于安装的是tensorflow2.x版本,会报错

解决办法是使用命令:! pip uninstall tensorflow

pip install tensorflow-gpu==1.14

pip install keras

10、运行代码有多种方式,可以在线编写,也可以本地执行。那么这里仅阐述使用colab上的GPU的方法。

首先上传需要训练的数据到与drive中该代码所在文件夹同一位置

右键->Upload files

在这里插入图片描述

然后打开所创建的.ipynb文件,这个文件的后缀名我的理解是:python+jupyter notebook

将本地所写的文件直接复制到该.ipynb文件的中一行中

在这里插入图片描述

最后更改训练数据路径,这个路径在哪呢?注意右边目录。点击
在这里插入图片描述

得到下图,依次点开,最后右键数据,复制路径,即可获得当前训练数据的路径,然后代码+数据+硬件都具备了,可以训练网络了。

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值