在pytorch0.3的時候，設備抽象還不具體，CPU設備要.cpu(), CUDA設備要.cuda(),切換起來就不太方便
到了pytorch 0.4和1.0版本以後，pytorch對所有的設備就都有了一個高層的抽象，不需要兩個不同的函數，直接用一個to(device)就行了
不管是cuda0還是cuda1還是CPU，GPU，都抽象成一個設備
如果有8張顯卡那你的cuda可以0-7編號
pytorch—GPU
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章