Tensorflow和Pytorch的框架安裝教程

這篇文章主要講解了“Tensorflow和Pytorch的框架安裝教程”，文中的講解內容簡單清晰，易于學習與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學習“Tensorflow和Pytorch的框架安裝教程”吧！

  在安裝深度學習框架前，首先需要安裝Anaconda3，cuda8.0和cudnn5.1。

1、安裝Anaconda3

Anaconda3可在官網上下載，或者去清華鏡像網站，地址是https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/，我選擇的是Anaconda3-4.2.0-Windows-x86_64版本。

下載完成后，我們直接按默認安裝即可，這個安裝過程比較費時間，大家耐心等待。

2、安裝cuda8.0和cudnn5.1

如果你的電腦沒有Nvidia顯卡的話，請直接跳過第2步。

cuda8.0下載地址是https://developer.nvidia.com/cuda-80-ga2-download-archive。下載完成后直接默認安裝即可。

  下載完成后，我們將三個文件分別放到 cuda的相應安裝目錄下。

4、安裝cpu和gpu版本的Pytorch

如果你的電腦是win10低配電腦，沒有Nvidia顯卡，那就只能安裝CPU版本的Pytorch了，在cmd中輸入：pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp35-cp35m-win_amd64.whl。

如果你的電腦是win10高配電腦，有Nvidia顯卡，你可以安裝CPU版本的Pytorch或安裝GPU版本的Pytorch，例如安裝GPU版本的Pytorch，在cmd中輸入：pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl。

感謝各位的閱讀，以上就是“Tensorflow和Pytorch的框架安裝教程”的內容了，經過本文的學習后，相信大家對Tensorflow和Pytorch的框架安裝教程這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是億速云，小編將為大家推送更多相關知識點的文章，歡迎關注！