一、查詢GPU的CUDA版本信息
- 查詢指令:``nvidia-smi`

GPU的CUDA版本信息
二、選用tensorflow-gpu版本
- 根據(jù)tensorflow_gpu-2.6.0下載CUDA11.2版本和cuDNN8.1版本。(Tensorflow對應版本)

tensorflow版本信息
三、安裝CUDA Toolkit
- 下載網址:(CUDA Toolkit 11.2)

CUDA 安裝路徑
四、安裝cuDNN8.1版本
下載網址:(cuDNN8.1)
cuDNN可以看作是CUDA的一個補丁,只要將對應文件拷貝進CUDA的對應文件夾((bin、include、lib))即可。

cuDNN下載頁面

CUDA補充文件

CUDA補充文件夾所在位置
五、效果測試
- 測試指令
nucc -V或nvcc --version

運行nvcc -V指令

運行.\bandwidthTest.exe指令

運行.\deviceQuery.exe指令
完成以上步驟,即可完成深度學習的運行環(huán)境搭建。接下來便是Anaconda和Tensorflow軟件的安裝。網上資料比較齊全,此處不再展開描述。