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

相关推荐
freewebsys2 分钟前
【原始POSTMAN】最老一个版本的postman 大家还记得吗?官方已经不支持没有升级了,下载了组件用AI直接升级了下可以使用了!
人工智能
云南新人和数采集团12 分钟前
云南靠谱软件开发公司选型指南:以昆明新腾科技为例
大数据·人工智能·科技
2601_967979121 小时前
Python第七十四讲 异常处理完全精通 try-except零基础保姆级教程
python·异常处理·编程教程·错误处理·try-except
发现你走远了1 小时前
05-权限不是开关:工具调用如何在放行、拦截与追问之间做决定
人工智能·agent·claude
AllData公司负责人1 小时前
AI大模型数据开发底座系列说明书:AllData集成DB-GPT开源项目建设 AI 多模态分析底座,涵盖模态解析/智能大屏/MCP 扩展!
大数据·数据库·人工智能·gpt·数据采集·数据接入·模态解析
wuyk5551 小时前
Spider-第07讲-JSON接口与正则提取
爬虫·python·json
2601_967987372 小时前
Sqlite参考手册
数据库·python·sqlite·api·教程
用户019027581612 小时前
A 股到价提醒机器人:AlphaFeed + 飞书/钉钉 Webhook
python
syyjhchou3 小时前
自己搭企业内部中转站,怎么让请求更容易命中缓存
人工智能·后端·语言模型·aigc
特立独行的猫A3 小时前
Ionic + 仓颉语言的 cj-tauri 开发框架实战:开发精美移动端 APP
java·后端·python