如何安装torch 1.5.0+cpu和torchvision 0.6.0+cpu

最新推荐文章于 2024-07-26 15:01:37 发布

菜菜码农一枚

最新推荐文章于 2024-07-26 15:01:37 发布

阅读量1w

点赞数 23

文章标签：神经网络机器学习 pytorch

本文链接：https://blog.csdn.net/Walter_fdh/article/details/106389388

版权

首先在PyTorch官网查看可用的最新版本，因笔记本显卡版本较低，安装cpu版本

可以看出，torch的最新版本为1.5.0，torchvision的最新版本为0.6.0. 鉴于pip和conda安装较慢，我们选择下载.whl文件半自动安装方式。从.whl下载地址中下载两个压缩文件，其中cpxx为python的版本（我的python是3.7.4）

torch 1.5.0+cpu

torchvision 0.6.0+cpu

将两者可以存在任何位置，为了便于管理存在下图的位置：

接下来就可以在自己的虚拟环境中安装了（也可以直接cmd中无目录指向安装）

pip install E:/Anaconda/myPkgs/torch-1.5.0+cpu-cp37-cp37m-win_amd64.whl -i http://pypi.mirrors.ustc.edu.cn/simple --trusted-host pypi.mirrors.ustc.edu.cn
pip install E:/Anaconda/myPkgs/torchvision-0.6.0+cpu-cp37-cp37m-win_amd64.whl -i http://pypi.mirrors.ustc.edu.cn/simple --trusted-host pypi.mirrors.ustc.edu.cn

很快就安装好啦！！！