26、利用GPU训练模型

1、使用GPU来训练，找到四个板块的cuda函数，进行调用

 

1、

 

 只有当cuda可用的时候才把变量引过去，下面调用的时候都一样，但是这样每次都写会很麻烦

2、数据集是不能直接调用cuda函数的

　　在训练和测试的时候调用

　　

 

 

 

3、损失函数调用cuda

 

 4、如果自己的电脑没有gpu，可以用谷歌的一个网站：Google colab提供免费的gpu（一周免费30个小时好像）

 

 

使用GPU时必须登录自己的谷歌账号才行

 进入网站之后-----点击新建笔记本-----修改选项中可以选择GPU