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

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)