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,安装成功。

相关推荐
___波子 Pro Max.8 分钟前
GitHub Actions配置python flake8和black
python·black·flake8
kngines10 分钟前
【字节跳动】数据挖掘面试题0007:Kmeans原理,何时停止迭代
人工智能·数据挖掘·kmeans
Kali_0713 分钟前
使用 Mathematical_Expression 从零开始实现数学题目的作答小游戏【可复制代码】
java·人工智能·免费
贾全20 分钟前
第十章:HIL-SERL 真实机器人训练实战
人工智能·深度学习·算法·机器学习·机器人
每日摸鱼大王25 分钟前
互联网摸鱼日报(2025-07-01)
人工智能
GIS小天35 分钟前
AI+预测3D新模型百十个定位预测+胆码预测+去和尾2025年7月4日第128弹
人工智能·算法·机器学习·彩票
我是小哪吒2.01 小时前
书籍推荐-《对抗机器学习:攻击面、防御机制与人工智能中的学习理论》
人工智能·深度学习·学习·机器学习·ai·语言模型·大模型
慕婉03071 小时前
深度学习前置知识全面解析:从机器学习到深度学习的进阶之路
人工智能·深度学习·机器学习
阿蒙Amon1 小时前
【Python小工具】使用 OpenCV 获取视频时长的详细指南
python·opencv·音视频
荔枝吻2 小时前
【AI总结】Git vs GitHub vs GitLab:深度解析三者联系与核心区别
人工智能·git·github