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版本了!

阅读更多
想对作者说点什么?

博主推荐

换一批

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