Pytorch安装指南（Linux/Win）

Prymce-Q

已于 2023-07-11 22:19:49 修改

阅读量1.2k

点赞数 1

文章标签： linux pytorch

于 2023-06-06 10:31:05 首次发布

本文链接：https://blog.csdn.net/weixin_47691066/article/details/131061368

版权

本文介绍了在Windows和Linux系统下安装PyTorch及其依赖Torchvision的步骤。首先，通过命令行检查CUDA和Python版本。在Windows上，以管理员身份运行CMD，从指定网站下载whl文件并使用pip安装。在Linux上，可以直接使用pip安装指定版本的PyTorch、Torchvision和torchaudio，也可以选择阿里云的镜像源加速下载。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前言

介绍如何在win/linux端安装pytorch及Torchvision。（这里跳过anaconda/cuda的安装）

先通过代码，确定cuda版本

nvcc --version

再确定python版本

python --version

一、Win

管理员身份启动cmd；
在该网站中 https://download.pytorch.org/whl/torch_stable.html 下载对应版本的torch和torchvision的whl；
使用下面的命令进行安装；

pip install [torch.whl]
pip install --no-deps [torchvision.whl]  # --no-deps，仅安装torchvision

二、Linux

pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 -f https://download.pytorch.org/whl/torch_stable.html

对应的版本可以在这个网址 https://download.pytorch.org/whl/torch_stable.html 里找。
可以 -i https://mirrors.aliyun.com/pypi/simple/ 加上镜像源来下载。