1.CUDA安裝(有N卡,v10.0)
什麼是CUDA
CUDA(ComputeUnified Device Architecture),是顯卡廠商NVIDIA推出的運算平臺。 CUDA是一種由NVIDIA推出的通用並行計算架構,該架構使GPU能夠解決複雜的計算問題。
下載安裝包
注意:tensorflow2.0對應的cuda版本是10.0,絕對不能下載錯。下載網頁鏈接
下載選項:
安裝步驟
下載好後打開安裝程序,安裝步驟如下:
設置如下
點擊下一步開始安裝。
之後就cuda就算安裝成功了。
2.安裝cuDNN(v7.5.0)
什麼是cuDNN
NVIDIA cuDNN是用於深度神經網絡的GPU加速庫。它強調性能、易用性和低內存開銷。NVIDIA cuDNN可以集成到更高級別的機器學習框架中,如谷歌的Tensorflow、加州大學伯克利分校的流行caffe軟件。CUDA看作是一個工作臺,而cuDNN就相當於工作的工具。想要在CUDA上運行深度神經網絡,就要安裝cuDNN。
下載安裝包
- 進入官網後,免費註冊一個賬號後就可以開始下載。
Download cuDNN v7.5.0 (Feb 21, 2019), for CUDA 10.0
解壓
下載好後,解壓得到名爲cuda的文件夾,然後將解壓縮後的文件複製到cuda安裝路徑下的cudnn文件夾(這個文件夾需要自己建)。如圖所示:
環境變量配置。
在系統變量的Path裏面添加下面的四個項目:缺一不可,且順序要在最上面(爲了避免與其他版本衝突):