ubuntu16.04安裝pytorch踩坑

本文只針對我在安裝過程中遇到的問題,僅供參考。具體安裝教程請參考pytorch官網(https://pytorch.org/get-started/locally/)。 第一次安裝:因爲電腦已經安裝了CUDA,所以直接按照官網的指令安裝,如下圖藍色框所示所示: 網速一定要好。顯示安裝成功之後,做測試。 爲確保正確安裝PyTorch,在這裏,我們將構造一個隨機初始化張量。如下圖所示,可以正確運行。
相關文章
相關標籤/搜索