如何安装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

这样你就可以下载成功了

相关推荐
Ronin-Lotus3 小时前
深度学习篇---剪裁&缩放
图像处理·人工智能·缩放·剪裁
毛飞龙3 小时前
Python类(class)参数self的理解
python··self
魔尔助理顾问3 小时前
系统整理Python的循环语句和常用方法
开发语言·后端·python
cpsvps4 小时前
3D芯片香港集成:技术突破与产业机遇全景分析
人工智能·3d
国科安芯4 小时前
抗辐照芯片在低轨卫星星座CAN总线通讯及供电系统的应用探讨
运维·网络·人工智能·单片机·自动化
AKAMAI4 小时前
利用DataStream和TrafficPeak实现大数据可观察性
人工智能·云原生·云计算
Ai墨芯1115 小时前
深度学习水论文:特征提取
人工智能·深度学习
无名工程师5 小时前
神经网络知识讨论
人工智能·神经网络
nbsaas-boot5 小时前
AI时代,我们更需要自己的开发方式与平台
人工智能
SHIPKING3935 小时前
【机器学习&深度学习】LLamaFactory微调效果与vllm部署效果不一致如何解决
人工智能·深度学习·机器学习