Pytorch学习笔记day1—— 安装教程

这里写自定义目录标题

工作需要,最近开始搞一点AI的事情。但是这个国产的AI框架,实话说对初学者不太友好
https://www.mindspore.cn/
比如说它不支持win下的CUDA,可是我手里只有3070Ti和4060也不太可能自己去买昇腾就有点绷不住。

嗯!但是!国产还是要支持滴!自家的孩子自己抱!遥遥领先!

所以!我们来先用一下pytorch,等逐渐熟悉了折腾linux环境吧。

Pytorch安装方式

安装还是挺简单的,首先你需要下载一个anaconda。我就随手官网上下载了一个最新的

官网贴一下 https://www.anaconda.com/

笔者用的是这个版本

anaconda和正常安装软件一样的这里略过,可能要注意一下不要搞中文路径。这个安装完就要安装Pytorch了

首先推荐一下这篇博客 https://blog.csdn.net/ccaoshangfei/article/details/126074300

然后给一个pytorch官网 https://pytorch.org/

pytorch会对python的版本以及cuda的版本比较敏感,所以要搞清楚自己当前的版本是啥

打开自己的Anaconda Prompt,我懒得搞一个独立环境就直接看了一下当前的python版本

cuda版本查看,桌面鼠标右键->NVIDIA控制面板->帮助->系统信息->组件

好,然后去pytorch选取合适的版本

把最后生成的这行贴到刚才的Anaconda Prompt里面

大概2G左右的东西,网络好一小会就安装好了。安装完以后,回到anaconda,打开python的ide,我这里用的是spyder

复制代码
import torch
 
# 测试GPU环境是否可使用
print(torch.__version__) # pytorch版本
print(torch.version.cuda) # cuda版本
print(torch.cuda.is_available()) # 查看cuda是否可用

运行结果如下,说明安装成功

相关推荐
K2_BPM几秒前
告别“单点智能”:AI Agent如何重构企业生产力与流程?
人工智能
TMT星球3 分钟前
深业云从人工智能产业投资基金设立,聚焦AI和具身智能相关产业
人工智能
哥布林学者11 分钟前
吴恩达深度学习课程四:计算机视觉 第二周:经典网络结构 (三)1×1卷积与Inception网络
深度学习·ai
鼾声鼾语23 分钟前
matlab的ros2发布的消息,局域网内其他设备收不到情况吗?但是matlab可以订阅其他局域网的ros2发布的消息(问题总结)
开发语言·人工智能·深度学习·算法·matlab·isaaclab
其美杰布-富贵-李30 分钟前
HDF5文件学习笔记
数据结构·笔记·学习
Dingdangcat8632 分钟前
中药材图像识别与分类 RetinaNet-R101-FPN模型详解
人工智能·数据挖掘
老蒋新思维34 分钟前
创客匠人视角:智能体重构创始人 IP,知识变现从 “内容售卖” 到 “能力复制” 的革命
大数据·网络·人工智能·tcp/ip·创始人ip·创客匠人·知识变现
Honmaple1 小时前
Spring AI 2.x 发布:全面拥抱 Java 21,Redis 史诗级增强
java·人工智能·spring
古城小栈1 小时前
区块链 + AI:医疗诊断模型存证上链技术实践与探索
人工智能·区块链
丹宇码农1 小时前
Index-TTS2 从零到一:完整安装与核心使用教程
人工智能·ai·tts