建站
咨詢
售后
建站咨詢
新聞中心
小編給大家分享一下在windows上如何安裝GPU版AI框架，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！
創(chuàng)新互聯(lián)專注骨干網(wǎng)絡(luò)服務(wù)器租用十余年，服務(wù)更有保障！服務(wù)器租用，服務(wù)器托管 成都服務(wù)器租用，成都服務(wù)器托管，骨干網(wǎng)絡(luò)帶寬，享受低延遲，高速訪問。靈活、實現(xiàn)低成本的共享或公網(wǎng)數(shù)據(jù)中心高速帶寬的專屬高性能服務(wù)器。
官方版本
安裝CUDA
安裝cuDNN
配置環(huán)境變量
安裝python環(huán)境
安裝gpu版的tensorflow開發(fā)包
咋看上去好像不是很復(fù)雜，但是其中坑多到你懷疑人生。
下載的cuDNN時候需要注冊，而且因為cuDNN文件在外網(wǎng)，下載速度很慢。
比如不同版本的tensorflow和CUDA（cuDNN）的版本是有匹配關(guān)系的，那你可能會說我先查好他們的這個依賴關(guān)系，然后找好合適的版本再安裝。
比如對于windows系統(tǒng)而言，顯卡驅(qū)動的版本決定了你能夠安裝的CUDA版本，有部分教程會忽略這一點。
......
假如你幸運地把上面的這些坑全部趟過了，那你還可能會遇到一個問題，就是如果你用pip去安裝tensorflow-gpu，還是有可能會報各種神奇的錯誤。事實上，在windows系統(tǒng)上安裝AI框架，最好的方法是用conda，而不是用原生的pip?？偨Y(jié)了上面的這些常見的問題，下面我們開始介紹如何在windows上極簡安裝GPU版AI框架：
極簡版本
安裝Anaconda
新建conda虛擬環(huán)境(建議用python3.6，可換國內(nèi)源)
安裝顯卡驅(qū)動（官網(wǎng)下載或者用驅(qū)動助手軟件）
安裝AI框架時同時安裝cudatoolkit包
結(jié)束啦
上面我們可以看到，這里我們使用了一個叫做cudatoolkit的包，相當于在conda的虛擬環(huán)境里面自動安裝了CUDA。大多數(shù)時候cudatoolkit中的CUDA和cuDNN是一起安裝的，這就省去了很多麻煩，比如說AI框架和CUDA的版本對應(yīng)問題以及進行手動設(shè)置環(huán)境變量等等。
具體的安裝腳本可參考如下命令：
安裝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框架”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對大家有所幫助，如果還想學(xué)習(xí)更多知識，歡迎關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道！
文章題目：在windows上如何安裝GPU版AI框架
分享URL：http://fisionsoft.com.cn/article/ggjsjh.html
