Pytorch安装详细步骤

第一步:检查显卡支持的的CUDA版本

1.打开NVIDIA控制面板

首先鼠标右击桌面-显示更多选项-NVIDIA控制面板-点击弹出界面右上角的(系统信息)-点击弹出界面的(组件)

2.查看驱动版本

打开系统信息

点击组件,查看

以观测到红色方框内的信息可以看到(NVIDIA CUDA 13.0.78 driver)就是我电脑支持CUDA的最高版本,可以下载12.x的版本,会稳定一些.(驱动如果不是最新的,可以更新,也可以根据驱动对应的cuda版本,选择对应cuda版本下载.)

第二步:安装CUDA

1.CUDA下载链接

:CUDA Toolkit Archive | NVIDIA Developer

2.下载CUDA安装包

打开界面如下, 我的电脑选择的是12.1的版本

点击后,出现如下界面,1.选择系统, 2.选择系统多少位的, 3.选择系统的版本,4 .安装方式选择本地

5.点击Download下载

3.安装CUDA

上面的地址是提取地址不用选择安装完后会自动删除,我们选择OK,经过一段时间加载后会弹出新的界面,我们 只需一路同意到安装选项的那一步时选择 自定义安装(高级)

选择自定义安装后点击下一步

按照操作对应的打钩即可,然后点击下一步进行安装会让我们选择安装地址(ps:一定要记住安装地址怕忘可以 先用记事本记住)这里建议选择默认地址即可省的之后出错。然后点击下一步等待安装完成即可。

第三步:安装cudnn

1.cudnn下载链接

cuDNN Archive | NVIDIA Developer

2.选择对应cuda版本下载

之前cuda下载的是12.1.0的版本,这里选择12.x的版本下载

windows的选择红色框下载即可.

3.下载之前先注册填写信息

没有账户的话,这里需要注册

信息填写后出现如下界面,选择是的,提交

提交后弹出如下界面,填写资料,

这个过程可能会有点折磨,他会不断的弹出填写资料的界面,最好英文状态下写,大学就写清华,北大这种,容易通过,语言记得选中文(简体)

如下,英文状态下填写,随便填一些信息就可以了,之后第2步

选择其中一个标签,完整配置文件后,即可下载.

下载后解压如下

将上面这几个文件复制到CUDA的安装目录下直接覆盖即可

第四步:在conda中创建pytorch_env虚拟环境

打开官网

点击Get started

可选历史版本

之后按如下操作即可:

1.打开应用中的Anaconda Prompt

  1. conda create --name tourch_env python=3.9使用如下命令创建虚拟环境,创建完成提示如下

3.使用conda activate tourch_env激活环境,之后就进入到该环境下

选择其中某个版本下载就好,但是cuda版本要选择12.1的,各自对应各自电脑上cuda 的版本,选择对应torch版本

复制命令,在激活的环境中,运行该命令,等待下载好即可

4.下载完成,进入python,import torch,运行,不报错,即安装成功!

相关推荐
Rock_yzh7 分钟前
AI学习日记——神经网络参数的更新
人工智能·python·深度学习·神经网络·学习
wa的一声哭了20 分钟前
Stanford CS336 assignment1 | Transformer Language Model Architecture
人工智能·pytorch·python·深度学习·神经网络·语言模型·transformer
waves浪游42 分钟前
Linux基本指令(中)
linux·运维·python
hui函数1 小时前
Python全栈(基础篇)——Day05:后端内容(dict与set+while循环+for循环+实战演示+每日一题)
开发语言·后端·python
haidizym1 小时前
ssc-FinLLM 金融大模型 相关链接
人工智能·算法
cxr8281 小时前
AI智能体赋能文化传承与创新领域:社群身份认同的数字空间重构与文化融合策略
大数据·人工智能·重构·提示词工程·ai赋能
常州晟凯电子科技1 小时前
海思SS626开发笔记之环境搭建和SDK编译
人工智能·笔记·嵌入式硬件·物联网
Apifox.1 小时前
Apifox 9 月更新| AI 生成接口测试用例、在线文档调试能力全面升级、内置更多 HTTP 状态码、支持将目录转换为模块
前端·人工智能·后端·http·ai·测试用例·postman
武子康1 小时前
AI-调查研究-95-具身智能 机器人场景测试全解析:从极端环境仿真到自动化故障注入
人工智能·深度学习·机器学习·ai·机器人·自动化·具身智能
Light602 小时前
领码方案|微服务与SOA的世纪对话(3):方法论新生——DDD、服务网格与AI Ops的融合之道
运维·人工智能·微服务·ddd·soa·服务网格·ai ops