二. pytorch環境安裝

Ubuntu 16.04環境 1.安裝nvidia 驅動+CUDA+cuDNN 對應版本安裝 測試CUDA nvidia-smi出現顯卡信息,則安裝成功。 2.安裝anaconda3 下載anaconda的sh文件,然後安裝 sudo bash ***.sh 3.安裝pytorch 一般使用conda intall 或者pip intall 注意匹配CUDA版本,舊版本點擊選擇 Previous
相關文章
相關標籤/搜索