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

相关推荐
岛雨QA2 小时前
【基础知识】人工智能大模型常见术语(1)
人工智能·aigc·openai
国家二级编程爱好者2 小时前
删除typora文档没有引用的资源文件
git·python
进击的雷神2 小时前
邮箱编码解码、国际电话验证、主办方过滤、多页面深度爬取——柬埔寨塑料展爬虫四大技术难关攻克纪实
爬虫·python
互联网志3 小时前
为数字经济与人工智能深度融合筑牢坚实底座
人工智能
l112206033 小时前
网站SEO定时自动发布工具优质厂家推荐
人工智能·推荐·seo优化·网站seo·定时发布工具·优质厂家
ASD123asfadxv3 小时前
RepVGG重参数化架构改进YOLOv26训练推理双模式与多分支融合协同突破
人工智能·yolo·目标跟踪
九硕智慧建筑一体化厂家3 小时前
一站式交付!绿色工厂节能降碳+照明+PLC集成解决方案直供
人工智能
深蓝电商API3 小时前
多线程 vs 异步 vs 多进程爬虫性能对比
爬虫·python
RuiBo_Qiu3 小时前
【LLM进阶-Agent】1.Agent范式发展历程
人工智能·ai-native
米小虾3 小时前
从 Prompt Engineering 到 Agent Engineering:AI Agent 架构设计与实践指南
人工智能