pytorch torchvision pip本地安装,可快速安装

pip 快速安装pytorch torchvision
1.先在命令窗口查看自己的cuda对应版本,如果有独立显卡,可使用cuda加速器处理,输入指令为nvidia-smi,我的版本为cuda11.4
在这里插入图片描述
2.可在cuda官网找到对应版本,参考其他专门安装cuda和cudnn的方式,安装好cuda和cudnn,我的已安装了cuda11.4。
3.安装pytorch,先进去pytorch官网:https://pytorch.org/get-started/locally/#windows-anaconda,选择自己电脑对应的pytorch版本,我的是cuda11.4,但官网最新为cuda11.3,所以我选了最新的,自己电脑cuda高一些没关系。我选了pip安装,对应最底部出现了安装指令:pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
在这里插入图片描述
4.当然可以安装这条指令安装,但是下载速度会很慢,时间久了容易报时间错误,因此,最快最稳定的方法就是,点击后面的链接,自己手动下载所需要的版本,https://download.pytorch.org/whl/cu113,点开链接如下图所示
在这里插入图片描述
5.可以点击自己想安装的模块,安装pytorch时,先安装torch,再安装torchvision。
**安装torch:**点击torch,可看到如下界面,我选择了我电脑对应版本,python3.7,当然没有cuda的,可选择cpu版本,(最好建立一个虚拟环境,conda creat pytorch python==3.7)
在这里插入图片描述

4.下载到本地后,命令窗口cd到所在文件夹,使用pip install torch-1.10.1+cu113-cp37-cp37m-win_amd64.whl,可实现pytorch的安装。
5.torchvision的安装过程类似,选择对应的版本即可,需要特别注意的是,使用pip install 安装时,会自动下载cpu的torch,卸载掉自己的cuda版本,解决办法是指令变为:pip install --no-deps 你的torchvision名称。但是这样安装后,导入会报错,缺一些库,比如numpy,pillow等,对应安装上就好。
6,安装库的时候,最好后面加镜像源,速度会更快
例如:pip install pillow -i https://pypi.tuna.tsinghua.edu.cn/simple/

我也试过conda install 安装方法,总会有问题,亲测这种方法既快又不会出错,祝大家好运!!!

  • 12
    点赞
  • 48
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值