python pytorch import error_pytorch安装 及 常见错误总结(长期更新)

pytorch安装

国内安装pytorch速度非常慢,建议离线实现离线下载好,保存起来

----------------------------------分割线----------------------------------

假定已经离线下载好目标版本whl文件

pip install torch.****.whl

conda install  torchvision=0.2.1 cudatoolkit=8.0

其中第二部conda install 可以事先选择清华源

常见问题

RuntimeError:CUDNN_STATUS_INTERNAL_ERROR

sudo rm -rf ~/.nv

https://github.com/SeanNaren/deepspeech.pytorch/issues/32

TypeError: expected Tensor as element 0 in argument 0, but got tuple

有一个变量不是tensor类型

python使用scipy.misc import imread报错:ImportError: cannot import name imread,imresize

很有可能当前安装scipy版本为1.3.1,此版本已经移除这几个函数,需要安装1.2.1

pip install scipy==1.2.1

ImportError: bad magic number in 'xxx': b'\x03\xf3\r\n'

可能是由于文件夹内存在.pyc文件,删除后重新运行即可

RuntimeError: set_sizes_contiguous is not allowed on Tensor created from .data or .detach()

pytorch版本问题,删除.data 或者 .detach()即可

即 a.data.b.c改为a.b.c

TypeError: cannot assign 'torch.cuda.FloatTensor' as parameter 'T' (torch.nn.Parameter or None expected)

错误示例如下

import torch

from torch.nn.parameter import Parameter

class MyLayer(nn.Module):

def __init__(self, inplace=True):

super(MyLayer, self).__init__()

'''others'''

self.T= Parameter(torch.Tensor(1))

'''others'''

def forward(self, input):

'''others'''

self.T = xxx

'''others'''

return xxx

根据提示可知,错误原因应该是类型不匹配,不可以将torch.cuda.FloatTensor赋值给Parameter类型,应该是赋值给Parameter的data属性,即

self.T.data = xxx

RCAN在pytorch1.1.0中运行错误:ImportError: cannot import name '_update_worker_pids',ImportError: cannot import name '_worker_manager_loop' from 'torch.utils.data.dataloader'

版本问题,解决方案参见【这里】

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值