CUDA,CUDNN的安装教程,让你可以使用Pytorch GPU显卡来跑程序


CUDA仅适用于有英伟达独立显卡电脑的同学.

1.CUDA的下载与安装

1.查看电脑支持的CUDA版本

按win+R打开cmd
输入 nvidia-smi可以看到你电脑最高支持的CUDA的版本
在这里插入图片描述

2.下载CUDA

打开链接: CUDA
选中你需要下载的版本
在这里插入图片描述
根据图示选择下载
在这里插入图片描述

3.CUDA的安装

双击打开后,这个是先对它进行解压,选用默认路径即可,安装完成后这个路径的文件会自动删除
在这里插入图片描述
解压完成会自动进入安装程序
点同意,下一步直到这里,这里第一次默认精简安装即可,安装时也会自动配置环境变量
在这里插入图片描述
最后安装完成

4.检验CUDA安装成功

打开cmd,输入 nvcc -V,输出以下信息即为安装成功
在这里插入图片描述
输入 set CUDA 即可看到CUDA的环境变量配置路径信息
在这里插入图片描述

2.下载安装CUDNN

1.下载CUDNN

打开链接 https://developer.nvidia.com/cudnn-downloads
我们需要先注册一个账号才能够下载
按图示所选下载即可
在这里插入图片描述

2.CUDNN的配置

下载后得到的是一个压缩包,我们将其解压,得到以下文件夹
在这里插入图片描述
将这三个文件夹都复制
在这里插入图片描述
打开下图所示的目录,并粘贴替换所有
在这里插入图片描述
在这里插入图片描述

3.验证是否安装成功

找到你安装目录的extras文件夹下的demo_suite文件夹,复制此目录

在这里插入图片描述

在cmd中输入 cd 此目录
在这里插入图片描述
分别输入下面这两行代码并允行

bandwidthTest.exe
deviceQuery.exe

在这里插入图片描述
在这里插入图片描述
最后出现PASS即为安装成功

3.Pytorch GPU 版本的安装

直接安装会torch会默认安装cpu版本的,想要让显卡干活还差最后一步
进入链接: Pytorch

在这里插入图片描述
往下滑到此页面,选中如上几条,会出来一个下载指令,我们将其复制
在需要下载pytorch的环境中运行这条指令即可

在这里插入图片描述
最后出来这条及下载成功GPU版本的pytorch
在这里插入图片描述

  • 10
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值