Pytorch-问题整理

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/pandamax/article/details/79961348

Q:RuntimeError: the given numpy array has zero-sized dimensions. Zero-sized dimensions are not supported in PyTorch
A:在pytorch 0.3版本中是不支持0维的,会出错.在pytorch 0.4 版本中解决了这个问题.但目前只能源码编译安装.condapip的安装方式官方还未给出.
验证版本命令:

>>> import torch
>>> print(torch.__version__)
0.4.0a0+2ef23b6

测试cuda:

torch.cuda.is_available()

________分割线________

目前pytorch 官方已经推出0.4版本了!

没有更多推荐了,返回首页

私密
私密原因:
请选择设置私密原因
  • 广告
  • 抄袭
  • 版权
  • 政治
  • 色情
  • 无意义
  • 其他
其他原因:
120
出错啦
系统繁忙,请稍后再试