安裝CUDA與cuDNN
1. 先檢查GPU版本
a. Device Manager->Display adaptors,如下圖所示
2. 檢查tensorflow所需要的CUDA及cuDNN版本
3. 安裝CUDA: https://developer.nvidia.com/cuda-toolkit-archive
所需版本請參照上圖
4. 下載cuDNN: https://developer.nvidia.com/rdp/cudnn-archive
需註冊Nvidia會員
解壓縮後將檔案複製到以下路徑:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
v11.0視個人所裝版本而定
5. 最後檢查cmd
1. ncvv -V
2. python->import tensorflow



沒有留言:
張貼留言