pytorch安装之本地安装torch和torchvision,并解决自动更新版本问题
vision
pytorch/vision: 一个基于 PyTorch 的计算机视觉库,提供了各种计算机视觉算法和工具,适合用于实现计算机视觉应用程序。
项目地址:https://gitcode.com/gh_mirrors/vi/vision

·
首先在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




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 天前
更多推荐
所有评论(0)