撰写于:  浏览:2269 次

1、PyTorch最新版本是多少?

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

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

2、PyTorch最新版本的安装

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

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

pip3 install torch torchvision torchaudio

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

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

(3)PyTorch 2.2.2 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

飞燕网专注于AI实习
合作伙伴有100+企业

    友情链接