在pytorch0.3的时候，设备抽象还不具体，CPU设备要.cpu(), CUDA设备要.cuda(),切换起来就不太方便
到了pytorch 0.4和1.0版本以后，pytorch对所有的设备就都有了一个高层的抽象，不需要两个不同的函数，直接用一个to(device)就行了
不管是cuda0还是cuda1还是CPU，GPU，都抽象成一个设备
如果有8张显卡那你的cuda可以0-7编号
pytorch—GPU
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章