在conda中安装PyTorch | pkgs中已有gpu pytorch的安装包,但仍无法使用gpu pytorch

今天跑模型时,发现自己之前安装的GPU pytorch用不了了,版本是1.7.0的。

>>> import torch
>>> torch.cuda.is_available()
False

不知道是什么原因。仔细回忆了一下,可能是因为之前自己 pip install 了 torchvision 工具包的缘故。

看了一下conda中存放python库的目录pkgs,发现有pytorch-1.8.0-py3.8_cuda10.2_cudnn7_0这么一个包的存在,这不就是GPU pytorch包吗?看了下这个文件的大小有2GB。那应该是安装了gpu pytorch的啊。

后来请教了一下某宝的专业人士,他的意思是因为我下载了cpu pytorch,把原来的gpu pytorch给盖掉了,只要卸载cpu pytorch即可。于是在anaconda prompt运行pip uninstall torch后,结果连import torch也开始报错了,找不到该包了…

后来想起自己之前是手动下载了1.7.0版本的gpu pytorch,安装包还留着。于是就重新安装了一下安装包。就是下面这个包。
在这里插入图片描述
运行

pip install 路径\torch-1.7.0+cu110-cp38-cp38-win_amd64.whl

安装完了以后,

>>> import torch
>>> torch.cuda.is_available()
True

于此,成功重装了gpu pytorch 。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值