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

相关推荐
AKAMAI1 小时前
跳过复杂环节:Akamai应用平台让Kubernetes生产就绪——现已正式发布
人工智能·云原生·云计算
databook2 小时前
Manim实现闪光轨迹特效
后端·python·动效
新智元2 小时前
阿里王牌 Agent 横扫 SOTA,全栈开源力压 OpenAI!博士级难题一键搞定
人工智能·openai
新智元3 小时前
刚刚,OpenAI/Gemini 共斩 ICPC 2025 金牌!OpenAI 满分碾压横扫全场
人工智能·openai
机器之心3 小时前
OneSearch,揭开快手电商搜索「一步到位」的秘技
人工智能·openai
阿里云大数据AI技术3 小时前
2025云栖大会·大数据AI参会攻略请查收!
大数据·人工智能
Juchecar3 小时前
解惑:NumPy 中 ndarray.ndim 到底是什么?
python
YourKing3 小时前
yolov11n.onnx格式模型转换与图像推理
人工智能
sans_3 小时前
NCCL的用户缓冲区注册
人工智能
sans_3 小时前
三种视角下的Symmetric Memory,下一代HPC内存模型
人工智能