安装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 可以用

相关推荐
科技圈快迅几秒前
商业旅拍后期修图软件实测:像素蛋糕功能与应用分析
人工智能
无忧智库13 分钟前
某矿山井下人员精准定位与AI行为安全识别管控系统建设方案(WORD)
人工智能·安全
CTA量化套保13 分钟前
一个账户跑多个期货策略:仓位与报单隔离思路
python·区块链
机汇五金_14 分钟前
影响交换机箱体使用寿命的几个关键因素
运维·服务器·网络·python
子午16 分钟前
基于DeepSeek的酒店客房管理系统~Python+DeepSeek智能问答+Vue3+Web网站系统
开发语言·前端·python
湘美书院--湘美谈教育17 分钟前
湘美谈教育AI经验集锦:有些东西,它们很难蒸馏
大数据·人工智能·深度学习·机器学习
HyperAI超神经21 分钟前
深度估计准确率冲上0.9,Meta提出VLM³,论证视觉模型天生会学3D,以Qwen3-VL-4B为基础实现多任务的统一建模
人工智能·3d·大模型·多模态·空间推理·3d感知·3d理解
编程大师哥21 分钟前
最高效的 IO 并发方案
linux·网络·python
Hello:CodeWorld22 分钟前
Dify 从入门到实战:部署、模型对接与企业级 AI 应用开发全教程
人工智能·python·架构·ai编程
本地化文档26 分钟前
black-docs-l10n
python·github·gitcode·sphinx