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

Windows 10下CUDA及cuDNN的安装

CUDA简介与下载地址

• CUDA(ComputeUnified Device Architecture)，是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题

• 下载网址：https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

验证：

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)