CUDA、CUDNN、Torch的配置

本文主要是介绍CUDA、CUDNN、Torch的配置，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

一、 配置CUDA

Win+R打开命令行，输入cmd，在终端输入：nvidia-smi。
查看本机的GPU信息

1.CUDA下载

CUDA下载链接：https://developer.nvidia.com/cuda-toolkit-archive
我机子支持 CUDA 12.5的，此处我以CUDA 12.5为例子。

2.CUDA安装

根据自己的需求选择即可。第一次安装时建议全选，就不用自己去配置cuda的环境。

3.CUDA配置环境

4.CUDA是否配置成功

Win+R打开命令行，输入cmd，在终端输入：ncvv -V
出现以下内容说明成功配置。

三、 配置CUDNN

CUDNN官方链接： https://developer.download.nvidia.cn/compute/cudnn/redist/cudnn/windows-x86_64/
下载对应的CUDA。

将下载后的文件复制到相应的文件夹中即可。

四、配置torch

在Anaconda中配置torch环境（方便管理）。

1.创建Python 3.8环境并激活

2.下载torch-GPU版本

Torch 官网链接：https://pytorch.org/get-started/locally/
官网下载的比较慢，可以同过镜像进行下载。


此处下载完即可。

这篇关于CUDA、CUDNN、Torch的配置的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---