提示:
当前PyTorch的最新版本是:PyTorch 2.8.0。
安装PyTorch 2.8.0需要的条件是:需要Python 3.9或者更高版本。
PyTorch安装分为两种方式:pip安装和whl安装。下面分别给大家介绍一下。
PyTorch的安装分为CPU版本和GPU版本,其安装命令如下所示:
(1)PyTorch 2.8.0 CPU版本的安装:
pip3 install torch torchvision torchaudio
(2)PyTorch 2.8.0 CUDA 12.6版本的安装:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
(3)PyTorch 2.8.0 CUDA 12.8版本的安装:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
(4)PyTorch 2.8.0 CUDA 12.9版本的安装:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu129
补充说明1:CUDA安装请移步:CUDA安装注意事项:CUDA的驱动版本和运行版本
(1)whl文件简介
whl格式本质上是一个压缩包,里面包含了py文件,以及经过编译的pyd文件。安装whl文件很简单,先下载whl文件,然后进入命令行,输入以下命令即可:pip install xxxx.whl
(2)PyTorch whl安装的必要性
当我们需要安装特定CUDA版本的PytTorch的时候,掌握whl文件的安装则是十分必要。
(3)PyTorch whl文件下载
PyTorch whl文件的下载地址:whl文件的下载
由于下载地址里面包含了太多的whl文件,让人眼花缭乱,常用的GPU版本的PyTorch文件如下所示:
CUDA 11.8版本的PyTorch文件请进入这个目录下面:CUDA 11.8文件下载
CUDA 12.4版本的PyTorch文件请进入这个目录下面:CUDA 12.4文件下载
CUDA 12.6版本的PyTorch文件请进入这个目录下面:CUDA 12.6文件下载
CUDA 12.8版本的PyTorch文件请进入这个目录下面:CUDA 12.8文件下载
CUDA 12.9版本的PyTorch文件请进入这个目录下面:CUDA 12.9文件下载