1. 程式人生 > 其它 >Windows 10下CUDA及cuDNN的安裝 —— Pytorch

Windows 10下CUDA及cuDNN的安裝 —— Pytorch

Windows 10下CUDA及cuDNN的安裝

CUDA簡介與下載地址

驗證:

cmd 視窗命令

nvcc -V

cudnn簡介與下載地址

  • NVIDIA cuDNN是用於深度神經網路的GPU加速庫。它強調效能、易用性和低記憶體開銷。NVIDIA cuDNN可以整合到更高級別的機器學習框架中,如谷歌的Tensorflow、加州大學伯克利分校的流行caffe軟體。簡單的插入式設計可以讓開發人員專注於設計和實現神經網路模型,而不是簡單調整效能,同時還可以在GPU上實現高效能現代平行計算。

  • 必須要註冊才能下載,但是註冊網頁打不開,則可以訪問 https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b
    在這個列表中選擇自己的版本 然後審查元素 用迅雷下載

重要步驟:複製貼上


其實是三個檔案 複製到對應的資料夾中

驗證:

import torch
print("ver:{}".format(torch.__version__))
print("cuda:{}".format(torch.cuda.is_available()))
print(torch.version.cuda)