也许没有察觉
在使用pytorch训练数据的时候cuda 显卡总是发挥不到最大性能
这就是你的cpu程序拖住了你的显卡
怎么办
目前我能想到的最好方法就是
使用numba.cuda.jit这样你也不用使用pytorch cuda
直接使用cuda.jit
这样所有的程序都是运行在cuda上面的
这样就不用考虑cuda等cpu的局面l
当然试试才知道这是一个想法而已
若使用numba.cuda.jit加速pytorch训练代码会怎样
最新推荐文章于 2022-06-30 21:44:18 发布