1. 程式人生 > 其它 >windows安裝cuda和cudnn

windows安裝cuda和cudnn

1、檢視windows系統支援的cuda的最高版本

2、安裝cuda

下載(這裡下載安裝11.1)

地址:https://developer.nvidia.cn/cuda-11.1.1-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

安裝

1、預設路徑

2、自定義安裝

3、選擇安裝項

4、編輯環境變數

5、成功安裝

3、安裝cudnn

1、下載

https://developer.nvidia.cn/rdp/cudnn-archive

2、複製檔案到

將解壓出來的檔案,複製到cuda對應的安裝路徑中

至此,cuda和cudnn安裝完畢

3、測試

用cmd從C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite目錄進入

1、執行deviceQuery.exe

2、執行bandwidthTest.exe

TensorFlow測試

import tensorflow as tf
tf.test.is_gpu_available()