PyTorch版本选择,安装,验证。

PyTorch 选择版本

打开Pytorch官网,

htps://pytorch.org/get-started/locally/Start Locallyhttp://htps://pytorch.org/get-started/locally/

根据需要选择要安装的PyTorch版本

在本例子中,这里选择最新的版本12.1生成的安装命令

bash 复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

注意,生成的安装指令默认是使用pip3,如果你用pip的改一下即可。

PyTorch 安装

PyTorch安装,开始执行安装,如下图:

提示安装完成,如下图:

验证PyTorch安装

验证PyTorch安装,执行下面python代码:

python 复制代码
import torch


if __name__ == '__main__':
    # 检测cuda环境
    print(torch.cuda.is_available())

验证结果为True,安装成功。

相关推荐
寒月霜华37 分钟前
机器学习-数据标注
人工智能·机器学习
九章云极AladdinEdu2 小时前
超参数自动化调优指南:Optuna vs. Ray Tune 对比评测
运维·人工智能·深度学习·ai·自动化·gpu算力
人工智能训练师3 小时前
Ubuntu22.04如何安装新版本的Node.js和npm
linux·运维·前端·人工智能·ubuntu·npm·node.js
酷飞飞3 小时前
Python网络与多任务编程:TCP/UDP实战指南
网络·python·tcp/ip
cxr8284 小时前
SPARC方法论在Claude Code基于规则驱动开发中的应用
人工智能·驱动开发·claude·智能体
研梦非凡4 小时前
ICCV 2025|从粗到细:用于高效3D高斯溅射的可学习离散小波变换
人工智能·深度学习·学习·3d
数字化顾问4 小时前
Python:OpenCV 教程——从传统视觉到深度学习:YOLOv8 与 OpenCV DNN 模块协同实现工业缺陷检测
python
幂简集成5 小时前
Realtime API 语音代理端到端接入全流程教程(含 Demo,延迟 280ms)
人工智能·个人开发
龙腾-虎跃5 小时前
FreeSWITCH FunASR语音识别模块
人工智能·语音识别·xcode
智慧地球(AI·Earth)5 小时前
给AI配一台手机+电脑?智谱AutoGLM上线!
人工智能·智能手机·电脑