【学习心得】安装cuda/cudann和pytorch

一、查看驱动信息

bash 复制代码
# 进入CMD输入命令
nvidia-smi

也可以右下角图标打开NVIDIA 设置进行查看

二、下载安装CUDA

1、下载

bash 复制代码
下载地址 https://developer.nvidia.com/

2、安装

推荐自定义安装。建议只勾选Cuda,只安装这一个就好,以免报错安装失败。

3、验证

验证CUDA是否安装成功,打开cmd,输入 nvcc --V,如下返回则说明安装成功。

三、下载安装CUDANN

1、下载

bash 复制代码
下载地址 https://developer.nvidia.com/rdp/cudnn-download


【注】Cuda和CudaNN的关系

cuda和cudann是nvidia用于GPU加速计算的两个关键技术。

它们的关系如下:

  • cuda是nvidia推出的gpu编程框架和平台,用于在gpu上进行通用计算。它提供了类似c语言的接口,使开发者可以使用cuda c/c++来编写gpu程序,实现GPU加速。
  • cudnn是nvidia基于cuda开发的深度神经网络加速库。它针对常见的神经网络运算进行了优化,可以充分利用gpu的并行计算能力,大幅加速神经网络的训练和推理过程。cudnn只支持nvidia gpu,需要安装cuda环境。

所以cudnn依赖于并扩展了cuda,专门用于加速深度学习领域的应用。它是nvidiacuda生态系统的重要组成部分。开发者可以建立在cuda基础上,利用cudnn进一步优化神经网络应用的性能。综上,cuda提供通用gpu计算框架,cudnn在此基础上进行深度学习优化。两者关系是:cudnn依赖cuda并进行扩展与优化,组成nvidia gpu加速平台的重要部分。

2、安装

解压,并把三个文件夹放在Cuda的安装路径下。就成功了。

3、验证

进入安装Cuda的根目录,找到extras/demo_suite文件夹。


在cmd中执行,分别执行deviceQuery和bandwidthTest,出现如下两个pass就说明成功了。

四、下载安装pytorch

1、下载

去pytorch.org官网,选择对应Cuda版本的pytorch版本。输入下面的命令进行安装。

2、安装

输入官网给你生成的pip命令就行

bash 复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

3、验证

bash 复制代码
# cmd中输入python进入交互模式
# 能导入成功说明有pytorch
import torch
# 显示pytorch的版本
torch.__version__
# 测试pytorch是否支持Cuda
torch.cuda.is_available()
# 看看Cuda操作的第0个显卡是什么(device就是显卡)
torch.cuda.device(0)
# 看看一共有几个显卡
torch.cuda.device_count()
# 显示第0个显卡(device)的名字
torch.cuda.get_device_name(0)

五、安装Cuda的坑

1、关闭电脑管家或者类似的杀毒软件

2、选择自定义安装

3、这几个不安装

建议只安装Cuda

bash 复制代码
Nsight开头的不安装
Visual Studio不安装
Driver components不安装(新版本小于当前版本就不安装)
NVIDIA GeForce Experience如果有也不安装

4、建议先根据pytorch版本选定cuda版本安装Cuda,再回过头来安装pytorch

相关推荐
麻雀无能为力27 分钟前
CAU数据挖掘实验 表分析数据插件
人工智能·数据挖掘·中国农业大学
时序之心31 分钟前
时空数据挖掘五大革新方向详解篇!
人工智能·数据挖掘·论文·时间序列
IMPYLH1 小时前
Python 的内置函数 reversed
笔记·python
.30-06Springfield1 小时前
人工智能概念之七:集成学习思想(Bagging、Boosting、Stacking)
人工智能·算法·机器学习·集成学习
说私域2 小时前
基于开源AI智能名片链动2+1模式S2B2C商城小程序的超级文化符号构建路径研究
人工智能·小程序·开源
永洪科技2 小时前
永洪科技荣获商业智能品牌影响力奖,全力打造”AI+决策”引擎
大数据·人工智能·科技·数据分析·数据可视化·bi
shangyingying_12 小时前
关于小波降噪、小波增强、小波去雾的原理区分
人工智能·深度学习·计算机视觉
小赖同学啊3 小时前
物联网数据安全区块链服务
开发语言·python·区块链
码荼3 小时前
学习开发之hashmap
java·python·学习·哈希算法·个人开发·小白学开发·不花钱不花时间crud
书玮嘎3 小时前
【WIP】【VLA&VLM——InternVL系列】
人工智能·深度学习