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

发布时间:2024年01月19日

?PyTorch 选择版本

打开Pytorch官网,

htps://pytorch.org/get-started/locally/Start Locallyicon-default.png?t=N7T8http://htps://pytorch.org/get-started/locally/

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

?

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

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

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

PyTorch 安装

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

提示安装完成,如下图:?

验证PyTorch安装

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

import torch


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

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

文章来源:https://blog.csdn.net/qyhua/article/details/135689451
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。