【学习心得】安装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

相关推荐
Fuly1024几秒前
prompt构建技巧
人工智能·prompt
on_pluto_2 分钟前
LLaMA: Open and Efficient Foundation Language Models 论文阅读
python·机器学习
小二·4 分钟前
mac下解压jar包
ide·python·pycharm
XXX-X-XXJ5 分钟前
二:RAG 的 “语义密码”:向量、嵌入模型与 Milvus 向量数据库实操
人工智能·git·后端·python·django·milvus
艾醒(AiXing-w)12 分钟前
探索大语言模型(LLM):大模型微调方式全解析
人工智能·语言模型·自然语言处理
科兴第一吴彦祖15 分钟前
基于Spring Boot + Vue 3的乡村振兴综合服务平台
java·vue.js·人工智能·spring boot·推荐算法
姚瑞南22 分钟前
【AI 风向标】四种深度学习算法(CNN、RNN、GAN、RL)的通俗解释
人工智能·深度学习·算法
antonytyler27 分钟前
认识机器学习
机器学习
渡我白衣44 分钟前
深度学习入门(一)——从神经元到损失函数,一步步理解前向传播(上)
人工智能·深度学习·学习
补三补四44 分钟前
SMOTE 算法详解:解决不平衡数据问题的有效工具
人工智能·算法