很多小伙伴在使用pytorch 指定驱动的时候会遇到两个功能类似的方法,就是to(device)和cuda(),这两种方法的功能都是指定CPU 或GPU 进行数据处理的 ...
確定! 回上一頁