1、使用GPU来训练,找到四个板块的cuda函数,进行调用
1、
只有当cuda可用的时候才把变量引过去,下面调用的时候都一样,但是这样每次都写会很麻烦
2、数据集是不能直接调用cuda函数的
在训练和测试的时候调用
3、损失函数调用cuda
4、如果自己的电脑没有gpu,可以用谷歌的一个网站:Google colab提供免费的gpu(一周免费30个小时好像)
使用GPU时必须登录自己的谷歌账号才行
进入网站之后-----点击新建笔记本-----修改选项中可以选择GPU
标签:26,调用,训练,模型,cuda,gpu,GPU From: https://www.cnblogs.com/ar-boke/p/17160300.html