PyTorch install 安装

撰写于:  浏览:3457 次

备注:本文更新于2024年8月12日

1、PyTorch最新版本是多少?

当前PyTorch的最新版本是:PyTorch 2.4.0。

安装PyTorch 2.4.0需要的条件是:需要Python 3.8或者更高版本。

2、PyTorch最新版本的安装

PyTorch的安装分为CPU版本和GPU版本,其安装命令如下所示:

(1)PyTorch 2.4.0 CPU版本的安装:

pip3 install torch torchvision torchaudio

(2)PyTorch 2.4.0 CUDA 11.8版本的安装:

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

(3)PyTorch 2.4.0 CUDA 12.1版本的安装:

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

补充说明1:CUDA安装

CUDA安装介绍请移步:http://www.hbase.cn/archives/1193.html

补充说明2:PyTorch whl安装

当我们需要安装特定CUDA版本的PytTorch的时候,掌握whl文件的安装则是十分必要。PyTorch whl文件的下载地址如下所示:
https://download.pytorch.org/whl/torch_stable.html

后记:这里最新版最全面最系统化的PyTorch教程,欢迎大家的关注:《PyTorch面试精华》

关注公众号,每天学一个AI知识

    友情链接