win10+anaconda3安装tensorflow-gpu和pytorch-gpu版
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、具體實現
2.1 windows10的安裝
在我之前寫過的安雙系統的一篇文章里面有提及安裝win10的教程在此不作過多贅述
2.2 Anaconda3的安裝(略)
安裝過程中提示“不建議添加到環境變量中”,那個默認不勾選就好,要啟動到開始里面,到Anaconda Prompt里面用管理員身份打開即可
2.3 Visual Studio 2017 community
現在官網上面已經提供最新的2019下載,之前的版本下載好像要訂閱什么的,你們也可以下載我這里提供的鏈接:https://pan.baidu.com/s/1Zvtg8nHPC_oZBA9rRC_CAA
提取碼:s90s ,安裝沒什么難的,主要之后的顯卡驅動要用到VS里面的東西所以要安裝它。
2.4 Nvidia drive 430.64的安裝
其實我感覺這個是不必須的,因為youtube上面很多教程都沒有提及這個,而且我發現后來安裝CUDA Toolkit的時候它也會裝drive,但是我是裝了,你們可以不裝試試看
去這里找你電腦的N卡對應的驅動,然后下載安裝,這個過程可能需要耐心等待,其中選項都默認就行
Nvidia驅動正在安裝
下面是我安好的截圖
顯卡驅動安裝成功
安好之后桌面上多了這樣一個圖標
驅動圖標
之后需要重啟(在我不重啟的時候后面的CUDA10.0安裝不成功),至此顯卡驅動已安裝完成!
?
2.5 CUDA Toolkt10.0的安裝
去這里下載對應版本的,我是win10本地安裝,如果不想安裝10.1可以在Release Notes里面找10.0版本的安裝
CUDA Toolkit10.0
這個是安裝中的截圖:
cuda10.0安裝過程圖
?
重啟后cuda10安裝成功
?
這里面都安裝在C盤,默認沒有改安裝路徑,安裝成功桌面上出現這樣的圖標:
cuda10.0安裝完圖標
2.6 cuDNN 7.5 for cuda10的安裝
去這里注冊下載,這個下載得先注冊,選擇cuDNN 7.5 for cuda10,因為我們之前裝的就是CUDA10.0
cuDNN7.5.1下載
下載完就是一個壓縮包
cudnn.png
,解壓出來就是一個名為“cuda”的文件夾,里面有
解壓出來的cuda中的文件
?
- 有些教程里面是說把這三個文件夾里面的.dll文件覆蓋掉cuda安裝目錄下的對應文件,我分析他們之后就沒有添加cuda的環境變量了,我沒試過
- 我是按照把C:\cuda\bin添加到環境變量里面了,我認為一樣的,我實踐證明有效
至此要安裝的東西幾乎完成了
3、 添加環境變量
除了上面說的把C:\cuda\bin添加到環境變量里面,還要照下圖的添加兩個環境變量
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp這就是剛剛CUDA的安裝目錄(當然上面是我自己的啦,你們要看自己的改一下)
4、安裝packages
4.1 安裝tensorflow-gpu版
進入到Anaconda Prompt里面(因為我沒有安裝python,這個是anaconda自帶的),因為anaconda包挺齊全的,所以直接安裝
pip3 install tensorflow-gpu我是下載下來.whl文件,然后cd到.whl目錄,用
pip install tensorflow_gpu-1.13.1-cp36-cp36m-win_amd64.whl這中間我出現過ModuleNotFoundError: No module named 'numpy.core._multiarray_umath' 錯誤,這個只要卸載numpy再重新安裝一下就好了
pip uninstall numpy pip install numpy下面我測試了一下,看看是否安裝成功了
?
測試tf安裝成功
4.2 安裝pytorch-gpu版
一般都是到官網去用一句話安裝,如
pytorch官網
但是這上面一般都是最新版的,點擊下面的Previous versions of PyTorch,可以找到對應平臺的.whl文件下載下來,和上面安裝tensorflow的.whl文件一樣安裝pytorch。
pytorch_gpu安裝成功
轉自簡書--賴子NEU
?
總結
以上是生活随笔為你收集整理的win10+anaconda3安装tensorflow-gpu和pytorch-gpu版的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: DualGan
- 下一篇: 图像处理/255.0 和/127.5 -