如何安装cuda版本的pytorch

为什么安装Cuda

对于做深度学习研究的小伙伴本,当我们处理大量的数据时,尤其是图像数据时,过量的数据会导致我们的CPU运行压力过大,占用大量的运行内存,而且用CPU进行模型训练,训练的时间会很长,这不利于我们PC用户的科学研究,相反,Cuda是NVIDIA开发出来的一个用于模型训练的一个很好的工具,可以快速有效的帮助我们进行模型的训练。

安装Cuda的流程

检查我们PC支持的Cuda版本

win + R 打开我们的运行管理器输入cmd,打开命令行,输入下面这行代码

bash 复制代码
nvidia -sim

我这里支持CUDA的版本是12.6

安装Cuda

这里附上官网链接

https://developer.nvidia.com/cuda-toolkit-archive

大家通过复制进行搜索

打开网址是这样的,你根据你支持的Cuda版本下载相应的Cuda

安装的过程一路点击确定即可,对于我们普通用户选择精简版就可以

Cuda 安装验证

同样的,在命令行中输入

bash 复制代码
nvcc -V

产生和下面这个图片差不多的信息就表明你安装成功了

Pytorch 的Cuda版本的安装

直接在我们的python终端执行

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

在whl后面加上你的cuda版本比如cuda11.8

这样你就可以下载成功了

相关推荐
weisian151几秒前
人工智能-基础篇-18-什么是RAG(检索增强生成:知识库+向量化技术+大语言模型LLM整合的技术框架)
人工智能·语言模型·自然语言处理
DataCastle6 分钟前
第三届Bio-OS AI开源大赛启动会隆重举行
人工智能
后端小肥肠15 分钟前
躺赚必备!RPA+Coze+豆包:公众号自动发文,AI率0%亲测有效(附AI率0%提示词)
人工智能·aigc·coze
破无差20 分钟前
python实现简单的地图绘制与标记20250705
python
摘星编程26 分钟前
CloudBase AI ToolKit实战:从0到1开发一个智能医疗网站
人工智能·腾讯云·ai代码远征季#h5应用·ai医疗应用·cloudbase开发
锅挤31 分钟前
深度学习5(深层神经网络 + 参数和超参数)
人工智能·深度学习·神经网络
一支烟一朵花35 分钟前
630,百度文心大模型4.5系列开源!真香
人工智能·百度·开源·文心一言
网安INF37 分钟前
深层神经网络:原理与传播机制详解
人工智能·深度学习·神经网络·机器学习
喜欢吃豆42 分钟前
目前最火的agent方向-A2A快速实战构建(二): AutoGen模型集成指南:从OpenAI到本地部署的全场景LLM解决方案
后端·python·深度学习·flask·大模型
好开心啊没烦恼1 小时前
Python 数据分析:DataFrame,生成,用字典创建 DataFrame ,键值对数量不一样怎么办?
开发语言·python·数据挖掘·数据分析