RuntimeError: Numpy is not available(关联Torch或Tensorflow)

RuntimeError: Numpy is not available解决方法

问题描述:

今日旧电脑崩(哭),新电脑上手,迫不及待地装好pytorch,查看之前的工程运行情况,结果运行模型训练时出错
出错位置极其奇怪
究其关键句,是加载模型训练数据的语句

for i, data in enumerate(train_loader):

该语句底层应用到了将tensor变量转换为numpy的操作,即

import numpy
import torch
test_torch = torch.rand(100,1,28,28)
print(test_torch.numpy())
# 将产生同样的错误

在这里插入图片描述

问题根源

忽略了Pytorch与Tensorflow的直接关联性。
如果环境中只有torch而没有tensorflow,那么torch使用到的tensor变量将失去tensor相关的操作函数,亦如torch.numpy()
在这里插入图片描述

解决方法

在基础环境中安装Tensorflow,无论是CPU版本还是GPU版本皆可。
在Anaconda环境中可以直接

conda install tensorflow

否则pip

pip install --ignore-installed --upgrade tensorflow-gpu
  • 25
    点赞
  • 30
    收藏
    觉得还不错? 一键收藏
  • 12
    评论
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值