安装CUDA Cudnn Pytorch(GPU版本)步骤

一.先看自己的电脑NVIDIA 支持CUDA版本是多少?

1.打开NVIDIA控制面板

2.点击帮助---系统信息--组件

我的支持CUDA11.6

二.再看支持Pytorch的CUDA版本

三.打开CUDA官网

下载CUDA 11.6

下载好后,安装

选择 自定义 然后安装位置 (先去F盘 建个CUDA-manger文件夹 然后在里面建个CUDA11.6文件夹 再在里面建立CUDA1 CUDA2这两个文件夹

前两个位置选到CUDA1

第三个选择CUDA2

然后下一步

最后验证 在anaconda promt里输入nvcc -V 可以看到CUDA的版本

四.然后去下载Cudnn,去官网

没有账号的需要注册一个,然后登陆,下载CUDA11.6对应的Cudnn,下载好之后,解压在F盘

五,安装Pytorch

找到对应的版本

看到CUDA 11.6 ,pytorch版本1.13.1 torchvision=0.14.1 torchaudio=0.13.1

如果 输入

复制代码
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

需要时间很久,好几个G ,所以我们还可以 去

复制代码
https://download.pytorch.org/whl/cu116

这个下载 torch torchaudio torchvision

把这三个文件 放在F:\whl里

然后在虚拟环境里pip install 这三个(我这里是F:\whl,对应的版本 上面给出 修改一下)

安装完之后,conda list 即可看见安装的库

验证:

输入python,进入解释器

输入import torch 返回>>> 说明可以导入 然后再torch.cuda.is_available() 返回True 就说明cuda 可以用

相关推荐
m4Rk_几秒前
【论文阅读】Agent记忆机制(112):MemFit——保留对话原文,用多路检索降低长期记忆成本
论文阅读·人工智能·学习·开源·github
AI日报派送佬1 分钟前
2026年10月21日AI行业日报|DeepSeek开源推理框架DeepInfer,谷歌TorchTPU开源破局TPU生态
人工智能·开源·openai·谷歌·英伟达·deepseek·ai日报
CAIE研习社2 分钟前
CAIE一级有学历和专业门槛吗?报考条件、考试内容与适合人群解析
大数据·人工智能
强哥聊算力6 分钟前
《算力江湖·DeepSeek》第四集:DeepSeek如何改变中国算力的采购逻辑?
大数据·运维·人工智能·ai·gpu算力
小雷信息医学11 分钟前
MedCortex:告别逐个试变量
人工智能
万化冥合16 分钟前
怎么批量把 Word 转 PDF?
python·pdf·c#·word
yh弓长18 分钟前
机器学习的四种范式
人工智能
QC777LX21 分钟前
金融企业选证书:专业资格是门槛,AI认证是补充
人工智能
~城南花已开~21 分钟前
Redfish 事件接收服务搭建指南(Ubuntu Server 实测)
linux·服务器·python·ubuntu
niucloud-admin22 分钟前
NIUCLOUD PHP SAAS上门家政使用文档
前端·人工智能·ai编程