1、需安裝的東西

• Win10 x64位
• Anaconda3 x64 python 3.6.5
• CUDA Toolkt10.0
• cuDNN 7.5 for cuda10
• Nvidia drive 430.64
• Visual Studio 2017 community

2、具體實(shí)現(xiàn)

2.1 windows10的安裝

在我之前寫過的安雙系統(tǒng)的一篇文章里面有提及安裝win10的教程在此不作過多贅述

2.2 Anaconda3的安裝（略）

安裝過程中提示“不建議添加到環(huán)境變量中”，那個(gè)默認(rèn)不勾選就好，要啟動(dòng)到開始里面，到Anaconda Prompt里面用管理員身份打開即可

2.3 Visual Studio 2017 community

現(xiàn)在官網(wǎng)上面已經(jīng)提供最新的2019下載，之前的版本下載好像要訂閱什么的，你們也可以下載我這里提供的鏈接：https://pan.baidu.com/s/1Zvtg8nHPC_oZBA9rRC_CAA
提取碼：s90s ，安裝沒什么難的，主要之后的顯卡驅(qū)動(dòng)要用到VS里面的東西所以要安裝它。

2.4 Nvidia drive 430.64的安裝

其實(shí)我感覺這個(gè)是不必須的，因?yàn)閥outube上面很多教程都沒有提及這個(gè)，而且我發(fā)現(xiàn)后來安裝CUDA Toolkit的時(shí)候它也會(huì)裝drive，但是我是裝了，你們可以不裝試試看
去這里找你電腦的N卡對(duì)應(yīng)的驅(qū)動(dòng)，然后下載安裝，這個(gè)過程可能需要耐心等待，其中選項(xiàng)都默認(rèn)就行

Nvidia驅(qū)動(dòng)正在安裝

下面是我安好的截圖

顯卡驅(qū)動(dòng)安裝成功

安好之后桌面上多了這樣一個(gè)圖標(biāo)

驅(qū)動(dòng)圖標(biāo)

之后需要重啟（在我不重啟的時(shí)候后面的CUDA10.0安裝不成功），至此顯卡驅(qū)動(dòng)已安裝完成！

?

2.5 CUDA Toolkt10.0的安裝

去這里下載對(duì)應(yīng)版本的，我是win10本地安裝，如果不想安裝10.1可以在Release Notes里面找10.0版本的安裝

CUDA Toolkit10.0

這個(gè)是安裝中的截圖：

cuda10.0安裝過程圖

?

重啟后cuda10安裝成功

?

這里面都安裝在C盤，默認(rèn)沒有改安裝路徑，安裝成功桌面上出現(xiàn)這樣的圖標(biāo)：

cuda10.0安裝完圖標(biāo)

2.6 cuDNN 7.5 for cuda10的安裝

去這里注冊(cè)下載，這個(gè)下載得先注冊(cè)，選擇cuDNN 7.5 for cuda10，因?yàn)槲覀冎把b的就是CUDA10.0

cuDNN7.5.1下載

下載完就是一個(gè)壓縮包

cudnn.png

，解壓出來就是一個(gè)名為“cuda”的文件夾，里面有

解壓出來的cuda中的文件

?

• 有些教程里面是說把這三個(gè)文件夾里面的.dll文件覆蓋掉cuda安裝目錄下的對(duì)應(yīng)文件，我分析他們之后就沒有添加cuda的環(huán)境變量了，我沒試過
• 我是按照把C:\cuda\bin添加到環(huán)境變量里面了，我認(rèn)為一樣的，我實(shí)踐證明有效

至此要安裝的東西幾乎完成了

3、 添加環(huán)境變量

除了上面說的把C:\cuda\bin添加到環(huán)境變量里面，還要照下圖的添加兩個(gè)環(huán)境變量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp

這就是剛剛CUDA的安裝目錄（當(dāng)然上面是我自己的啦，你們要看自己的改一下）

4、安裝packages

4.1 安裝tensorflow-gpu版

進(jìn)入到Anaconda Prompt里面（因?yàn)槲覜]有安裝python，這個(gè)是anaconda自帶的），因?yàn)閍naconda包挺齊全的，所以直接安裝

pip3 install tensorflow-gpu

我是下載下來.whl文件，然后cd到.whl目錄，用

pip install tensorflow_gpu-1.13.1-cp36-cp36m-win_amd64.whl

這中間我出現(xiàn)過ModuleNotFoundError: No module named 'numpy.core._multiarray_umath' 錯(cuò)誤，這個(gè)只要卸載numpy再重新安裝一下就好了

pip uninstall numpy pip install numpy

下面我測(cè)試了一下，看看是否安裝成功了

?

測(cè)試tf安裝成功

4.2 安裝pytorch-gpu版

一般都是到官網(wǎng)去用一句話安裝，如

pytorch官網(wǎng)

但是這上面一般都是最新版的，點(diǎn)擊下面的Previous versions of PyTorch，可以找到對(duì)應(yīng)平臺(tái)的.whl文件下載下來，和上面安裝tensorflow的.whl文件一樣安裝pytorch。

pytorch_gpu安裝成功

轉(zhuǎn)自簡(jiǎn)書--賴子NEU
?

總結(jié)

以上是生活随笔為你收集整理的win10+anaconda3安装tensorflow-gpu和pytorch-gpu版的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。