![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
weixin_44602056
这个作者很懒,什么都没留下…
展开
-
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED using pytorch
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED using pytorchtest codeimport torchfrom torch import nnm = nn.Conv1d(16, 33, 3, stride=2)m=m.to('cuda')input = torch.randn(20, 16, 50)input=input.to('cuda')output = m(input)I guess this is no原创 2021-03-17 10:56:36 · 506 阅读 · 0 评论 -
pip安装torch, torchvision, opencv-python太慢的问题解决
http://download.pytorch.org/whl/torch_stable.html去这个网址: torch下载下载所需要的torch.whl and torchvision.whl 然后单独安装即可,因为pip实在是太慢了。原创 2021-03-16 15:14:10 · 3253 阅读 · 1 评论