首先在pytorch 官网下载对应版本的.whl文件

官网链接:https://download.pytorch.org/whl/torch_stable.html
在这里插入图片描述
找到自己需要的版本,下载需要十几分钟。

下载完成后,打开anaconda prompt,进入虚拟环境,这里以名为“pipelearn”的环境为例,

conda activate pipelearn

使用pip安装,我的whl文件名称为torch-2.0.0+cu118-cp39-cp39-win_amd64.whl

pip install torch-2.0.0+cu118-cp39-cp39-win_amd64.whl

在这里插入图片描述
安装成功后如图

安装torchvision

使用pip安装,这里要注意默认的安装为最新版本,这会导致上一步安装的torch自动更新,可能会引起与gpu版本不一致,所以需要指定torchvision的安装版本。版本对应如下
在这里插入图片描述
在这里插入图片描述
以我需要的版本为例

pip install torchvision==0.15.1

在这里插入图片描述
最后查看是否安装完成

conda list

在这里插入图片描述

GitHub 加速计划 / vi / vision
34
0
下载
pytorch/vision: 一个基于 PyTorch 的计算机视觉库,提供了各种计算机视觉算法和工具,适合用于实现计算机视觉应用程序。
最近提交(Master分支:23 天前 )
7a13ad0f Differential Revision: D82494990 Pull Request resolved: https://github.com/pytorch/vision/pull/9231 5 天前
58eb0393 Co-authored-by: Nicolas Hug <nicolashug@meta.com> 6 天前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐