Linux下Pytorch1.1.0(GPU)安裝筆記

1.cuda, pytorch版本對應問題     設備上的GPU驅動版本決定了cuda版本,而cuda版本與cudatoolkit的版本一致, 而pytorch(GPU)的版本要與cudatoolkit的版本嚴格一致。      一開始沒留意這個版本的對應關係,直接輸入官網上給出的命令,踩了大坑!(官網上的cudatoolkit版本爲9.2,而我設備上的cudatoolkit版本爲9.0,不對應
相關文章
相關標籤/搜索