- 博客(3)
- 收藏
- 关注
原创 关于torch.size()和torch.size(0)
在分类模型测试代码中有以下代码段:correct = 0total = 0# since we're not training, we don't need to calculate the gradients for our outputswith torch.no_grad(): for data in testloader: images, labels = data # calculate outputs by running images thr
2021-08-19 00:38:57 5330 1
原创 解决OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.
问题:Anaconda安装的numpy包中的libiomp5md.dll 和Pytorch中libiomp5md.dll冲突解决方法:添加环境变量: import os os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE" 降低numpy 版本 pip install -U numpy==1.16.5 Done!
2021-08-18 14:45:25 1488 2
原创 Windows编译mmcv-full报OSError: [WinError 126]
根据WINDOWS 下 MMCV | MMCV-full 的安装安装mmcv-full的过程中在进行以下这一步时python setup.py build_ext报OSError:[WinError 126]但实际上torch_cuda_cpp.dll路径是正确的而后尝试用cmd import torch 也出现了该错误,也是找不到torch_cuda_cpp.dll这个文件:在尝试修改__init__文件、安装vc_redist.x64.exe等操作后问题还是没有解决。后
2021-07-20 00:25:47 739
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人