ubuntu18.04下安裝pytorch-gpu版

前面 之前介紹過TensorFlow-gpu版本的安裝過程。現在再說說pytorch的安裝過程。 1.安裝Ubuntu後，先安裝nvida顯卡驅動。 2.安裝anaconda。 3.安裝cuda。 4.安裝cudnn 5.安裝pytorch-gpu 前面5個步驟 參考 TensorFlow的安裝過程中。 現在主要介紹下pytorch-gpu的安裝。 5、安裝pytorch-gpu 進入官網 選擇對

>>阅读原文<<