windows安装pytorch

windows安装pytorch

通过pip来安装pytorch.

1、更新pip

在激活的虚拟环境中，输入命令：

python -m pip install --upgrade pip -i mirrors.aliyun /pypi/simple/ 2、安装cpu版的pytorch pip3 install torch torchvision torchaudio -i mirrors.aliyun /pypi/simple/ 3、安装cuda pip3 install torch torchvision torchaudio --index-url download.pytorch.org/whl/cu118 -i mirrors.aliyun /pypi/simple/

替换cu118为你的CUDA版本（如果适用）。