- 到pytorch官网复制安装命令Start Locally | PyTorch
- 以下以cuda11.8+python3.8为例
- 执行命令,如
pip install torch==2.5.0+cu124 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
复制控制台输出的下载路径如:download.pytorch.org/whl/cu118/t… 到迅雷进行下载文件 下载完成后安装文件
pip install D:\迅雷下载\torch-2.4.0+cu118-cp38-cp38-win_amd64.whl
重装运行官网复制的安装命令进行安装剩余两个大小较小的文件完成pytorch安装过程
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
检查是否安装成功
import torch
print(torch.__version__)
输出为2.4.0+cu118代表安装成功