pyTorch
upDiff
专注于计算机视觉和深度学习领域,8年+AI项目实战经验。善于将复杂的概念简化为易于理解的形式,并提供清晰的示例代码和运行结果,帮助大家速掌握相关技术。
展开
-
pytorch报错“some of the strides of a given numpy array are negative“
在运行pytorch程序时,出现了如下的错误:经过查找资料,发现是Opencv读取图片时的原因,原始报错代码片段:img = cv2.imread(imgpath)img = img[:,:,::-1]img_tensor = torch.from_numpy(img)修改后的代码:img = cv2.imread(imgpath)img = img[:,:,::-1].copy()img_tensor = torch.from_numpy(img)...原创 2020-08-13 15:40:01 · 443 阅读 · 0 评论 -
pytorch多GPU并行以及注意事项
pytorch多GPU并行以及注意事项pytorch多GPU并行方式1-参数解析2-使用数据并行3-保存权重4-运行程序5-注意事项pytorch多GPU并行方式1-参数解析使用argparse进行参数解析,核心参数如下:import argparseparser.add_argument("--local_rank", default=0, type=int)parser.add_argument("--ngpu", default=6, type=int)args.gpu = args.l原创 2020-08-03 17:49:46 · 658 阅读 · 0 评论 -
conda安装cpu版pytorch
conda install pytorch-cpu=1.0pip install torchvision==0.2 -i https://pypi.tuna.tsinghua.edu.cn/simple原创 2020-05-11 19:57:20 · 1422 阅读 · 0 评论 -
AttributeError: module 'torch' has no attribute 'flip'
运行基于pytorch写的程序时出现"AttributeError: module ‘torch’ has no attribute ‘flip’"的错误提示,修改方法:将pytorch版本升级到1.0以上即可原创 2020-04-29 11:16:15 · 1080 阅读 · 0 评论 -
TypeError: ‘_IncompatibleKeys‘ object is not callable
在使用pytorch编写代码时,出现了"TypeError: ‘_IncompatibleKeys’ object is not callable",报错原因如下:model = model.load_state_dict(torch.load('model.pt'))model.load_state_dict(torch.load(‘model.pt’))返回结果不是model类型,正确写...原创 2020-04-21 14:14:40 · 2941 阅读 · 0 评论 -
RuntimeError: Ninja is required to load C++ extension
wget https://github.com/ninja-build/ninja/releases/download/v1.8.2/ninja-linux.zipsudo unzip ninja-linux.zip -d /usr/local/bin/sudo update-alternatives --install /usr/bin/ninja ninja /usr/local/bin...原创 2019-09-30 09:53:42 · 4355 阅读 · 1 评论 -
apex安装
安装apex步骤如下,git clone https://github.com/NVIDIA/apex.gitcd apexpython setup.py install --cpp_ext --cuda_ext参考链接:https://github.com/NVIDIA/apex/issues/156原创 2019-06-25 15:05:14 · 16047 阅读 · 4 评论 -
PyTorch节省显存占用方法
1-使用inplace操作2-使用混合精度运算参考:[1]混合精度训练http://kevinlt.top/2018/09/14/mixed_precision_training/[2]pytorch节省显存小技巧https://blog.csdn.net/yangyang_yangqi/article/details/88051462[3]Pytorch有什么节省内存...原创 2019-06-21 08:17:28 · 1943 阅读 · 0 评论 -
conda安装pytorch后没有找到cudnn
使用命令conda install pytorch torchvision cudatoolkit=10.0 -c pytorch安装pytorch后,用conda list找不到cudnn,原因是pytorch携带cudnn,检查方法如下:import torchprint(torch.backends.cudnn.enabled)...原创 2019-04-01 10:43:01 · 5392 阅读 · 6 评论 -
pytorch改变tensor的类型
假设变量名为ts方法一:ts=ts.double()ts=ts.long()方法二:ts=ts.type(torch.DoubleTensor())ts=ts.type(torch.LongTensor())原创 2019-03-04 10:22:16 · 7806 阅读 · 0 评论 -
pytorch程序异常后删除占用的显存
1-删除模型变量del model_define2-清空CUDAcachetorch.cuda.empty_cache()3-步骤2(异步)需要一定时间,设置时延time.sleep(5)完整代码如下:del styler torch.cuda.empty_cache() time.sleep(5)...原创 2019-01-29 11:44:14 · 5489 阅读 · 0 评论 -
from torch._C import * ImportError: numpy.core.multiarray failed to import
安装pyTorch时,按照官网的命令进行安装pip3 install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp35-cp35m-manylinux1_x86_64.whl pip3 install torchvision‘import torch’会抛出”from torch._C import * ImportError:原创 2017-11-20 17:36:48 · 7321 阅读 · 2 评论