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是英伟达开发出的GPU计算框架。英伟达是制造GPU的公司,它为了让人们更好地使用自家生产出来的GPU,所以开发出CUDA用于操作GPU。如果没有CUDA,直接操作GPU则是一件非常困难的事情。
我们买回的带英伟达GPU的新电脑,会自带GPU显卡驱动,显卡驱动的版本号制约着你将来要安装的CUDA。运行nvidia-smi
命令,我们可以查看自己显卡的驱动版本:
然后,根据显卡驱动版本,到英伟达官网,查找相应的CUDA开发工具包:
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
安装完CUDA开发工具包,运行nvcc -V
:
需要注意的是:运行nvidia-smi
命令出现的窗口中也出现了CUDA版本号,如下所示:
这是画蛇添足,反而给人们造成了很大的困惑。它指的是在当前显卡驱动的版本下,可以支持的CUDA的最大版本。很多时候,当我们第一次执行nvidia-smi
的时候,电脑上还并没有安装CUDA呢,所以大家务必要知道,这个窗口出现的CUDA版本号不是你所安装的CUDA的版本号。
补充说明2:PyTorch whl安装
当我们需要安装特定CUDA版本的PytTorch的时候,掌握whl文件的安装则是十分必要。PyTorch whl文件的下载地址如下所示:
https://download.pytorch.org/whl/torch_stable.html