Windows 10下CUDA及cuDNN的安裝 —— Pytorch
阿新 • • 發佈:2021-10-17
Windows 10下CUDA及cuDNN的安裝
CUDA簡介與下載地址
-
CUDA(ComputeUnified Device Architecture),是顯示卡廠商NVIDIA推出的運算平臺。 CUDA是一種由NVIDIA推出的通用平行計算架構,該架構使GPU能夠解決複雜的計算問題
驗證:
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)