pytorch
ZQSZXY
这个作者很懒,什么都没留下…
展开
-
RuntimeError: CUDA error: invalid device ordinal (exchangeDevice at /opt/conda/conda-bld/pytorch_154
pytorch出现下面的错误RuntimeError: CUDA error: invalid device ordinal (exchangeDevice at /opt/conda/conda-bld/pytorch_1549628766161/work/aten/src/ATen/cuda/detail/CUDAGuardImpl.h:28)很大可能是你GPU参数设置的不正确。比如你只有...原创 2019-11-13 20:38:10 · 1697 阅读 · 0 评论 -
pytorch中torch.cat()函数用法
torch.cat(seq,dim,out=None)其中seq表示要连接的两个序列,以元组的形式给出,例如:seq=(a,b), a,b 为两个可以连接的序列dim 表示以哪个维度连接,dim=0, 横向连接,dim=1,纵向连接举例如下:import torcha = torch.ones([1, 2])b = torch.ones([1, 2])print(torch.cat...原创 2019-02-14 11:12:51 · 13496 阅读 · 1 评论 -
undefined symbol: _ZN3c1019UndefinedTensorImpl10_singletonE
ImportError: /usr/local/lib/python3.5/dist-packages/torchvision/_C.cpython-35m-x86_64-linux-gnu.so: undefined symbol: _ZN3c1019UndefinedTensorImpl10_singletonE原创 2019-06-16 17:43:20 · 9222 阅读 · 1 评论