【環境配置】CUDA和cuDNN安裝

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。

下載安裝包

解壓

下載好後,解壓得到名爲cuda的文件夾,然後將解壓縮後的文件複製到cuda安裝路徑下的cudnn文件夾(這個文件夾需要自己建)。如圖所示:
在這裏插入圖片描述

環境變量配置。

在系統變量的Path裏面添加下面的四個項目:缺一不可,且順序要在最上面(爲了避免與其他版本衝突):
![在這裏插入圖片描述](https://img-blog.csdnimg.cn/20200301151217549.png

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章