在windows上如何安裝GPU版AI框架

小編給大家分享一下在windows上如何安裝GPU版AI框架，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

官方版本

1. 安裝CUDA

2. 安裝cuDNN

3. 配置環境變量

4. 安裝python環境

5. 安裝gpu版的tensorflow開發包

咋看上去好像不是很復雜，但是其中坑多到你懷疑人生。

• 下載的cuDNN時候需要注冊，而且因為cuDNN文件在外網，下載速度很慢。

• 比如不同版本的tensorflow和CUDA（cuDNN）的版本是有匹配關系的，那你可能會說我先查好他們的這個依賴關系，然后找好合適的版本再安裝。

• 比如對于windows系統而言，顯卡驅動的版本決定了你能夠安裝的CUDA版本，有部分教程會忽略這一點。

• ......

  假如你幸運地把上面的這些坑全部趟過了，那你還可能會遇到一個問題，就是如果你用pip去安裝tensorflow-gpu，還是有可能會報各種神奇的錯誤。事實上，在windows系統上安裝AI框架，最好的方法是用conda，而不是用原生的pip。總結了上面的這些常見的問題，下面我們開始介紹如何在windows上極簡安裝GPU版AI框架：

極簡版本

1. 安裝Anaconda

2. 新建conda虛擬環境(建議用python3.6，可換國內源)

3. 安裝顯卡驅動（官網下載或者用驅動助手軟件）

4. 安裝AI框架時同時安裝cudatoolkit包

5. 結束啦

上面我們可以看到，這里我們使用了一個叫做cudatoolkit的包，相當于在conda的虛擬環境里面自動安裝了CUDA。大多數時候cudatoolkit中的CUDA和cuDNN是一起安裝的，這就省去了很多麻煩，比如說AI框架和CUDA的版本對應問題以及進行手動設置環境變量等等。

具體的安裝腳本可參考如下命令：

安裝tensorflow

conda create -n py36 python=3.6

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --set show_channel_urls yes

conda install cudatoolkit tensorflow-gpu

安裝pytorch

conda create -n py36 python=3.6

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --set show_channel_urls yes

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

以上是“在windows上如何安裝GPU版AI框架”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！