一、查詢GPU的CUDA版本信息
- 查詢指令：``nvidia-smi`
GPU的CUDA版本信息
二弧轧、選用tensorflow-gpu版本
- 根據(jù)tensorflow_gpu-2.6.0下載CUDA11.2版本和cuDNN8.1版本宰翅。（Tensorflow對應版本）
tensorflow版本信息
三账阻、安裝CUDA Toolkit
- 下載網(wǎng)址:（CUDA Toolkit 11.2）
CUDA 安裝路徑
四吵聪、安裝cuDNN8.1版本
下載網(wǎng)址：（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軟件的安裝纷闺。網(wǎng)上資料比較齊全，此處不再展開描述贬派。
