pytorch
关注
分享
扫一扫
文章平均质量分 84
全世界最为流行的深度学习框架有Pytorch、PaddlePaddle、Tensorflow、Caffe，这个专栏整理我自己的pytorch博客，只为了当个笔记，并无它意
知识在于分享
即便是蜗牛，一步一步不停歇的爬也会有看见太阳的一天
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
ImportError: libcusparse.so.10.0: cannot open shared object file: No such file or directory
一般出现这种情况,肯定是使用pycharm,然后他找不到cuda虽然你在~/.bashrc 中已经设置好了:export CUDA_HOME=/usr/local/cudaexport PATH=/usr/local/cuda/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH但是还是会出问题,因为pycharm并没有关联到网上有一些建议修改pycharm.sh我觉得不妥,万一想换cuda怎么办原创 2020-09-24 11:08:42 · 3264 阅读 · 0 评论 -
【个人笔记】深度学习模型部署--2019.8.6下一步学习计划
搭建深度学习后台服务器https://blog.youkuaiyun.com/qq_38957170/article/details/82820541本篇文章的原创为国外的一篇文章(一个可扩展的Keras深度学习REST API)，链接为：https://www.pyimagesearch.com/2018/01/29/scalable-keras-deep-learning-rest-api/...原创 2019-08-06 13:10:35 · 321 阅读 · 0 评论 -
mmdetection工具网上博客整理
安装：(pytorch1_1_0--py3_6--cuda9_0) spple@spple:~/paddle/vis-project/mmdetection/mmdetection$ python setup.py develop/home/spple/anaconda3/envs/pytorch1_1_0--py3_6--cuda9_0/lib/python3.6/site-packag...转载 2019-08-04 22:15:05 · 2736 阅读 · 1 评论 -
pytorch转caffe步骤
https://blog.youkuaiyun.com/vvnzhang2095/article/details/91439924tensorflow ckpt文件转caffemodel时遇到的坑https://www.cnblogs.com/unclelin/p/6984436.html转载 2019-07-24 13:39:17 · 447 阅读 · 0 评论 -
caffe-tensorflow-darknet-pytorch关于pad的比较
Tnesrflow：https://www.imooc.com/article/details/id/29525https://blog.youkuaiyun.com/wuzqChom/article/details/74785643https://www.cnblogs.com/White-xzx/p/9497029.html3.卷积padding的实战分析接下来我们通过在Tenso...转载 2019-07-24 13:27:32 · 1164 阅读 · 0 评论 -
pytorch1.1.0-python3.6-CUDA9.0-以各种方式打印模型
CUDA模式下，当然也可以cpu模式下import tensorwatch as twdummy_input = torch.randn(1, 1, 32, 32, device='cuda')tw_graph = tw.draw_model(sosnet32, dummy_input)tw_graph.save("./mode_1.pdf")import hiddenlaye...原创 2019-06-24 16:46:51 · 1600 阅读 · 0 评论 -
pytorch文档地址与ONNX算子文档
https://pytorch.apachecn.org/https://github.com/onnx/onnx/blob/master/docs/Operators.md原创 2019-06-18 15:45:14 · 3398 阅读 · 1 评论 -
快速入门PyTorch
转载 2019-06-18 13:49:56 · 244 阅读 · 0 评论 -
pytorch1.1.0-python3.6-CUDA9.0-ToTensor ToPILImage
参考：PIL.Image和np.ndarray图片与Tensor之间的转换https://blog.youkuaiyun.com/tsq292978891/article/details/78767326Pytorch之深入torchvision.transforms.ToTensor与ToPILImagehttps://blog.youkuaiyun.com/qq_37385726/article/de...转载 2019-06-14 21:49:31 · 626 阅读 · 0 评论 -
pytorch1.1.0-python3.6-CUDA9.0-以10分类为例子，使用自建类载入数据，训练网络(efficientnet)
资料来源于网络，但是修改了一些东西，比如载入数据，可以灵活的读取，自定路径+txt内写入的路径： # 构建MyDataset实例 img_path是一种可在txt图片路径前面加入的一种机制 train_data = MyDataset(img_path = '', txt_path=train_txt_path, transform=trainTransform) va...原创 2019-06-14 20:52:48 · 2152 阅读 · 1 评论 -
pytorch1.1.0-python3.6-CUDA9.0-以MNIST为例子，使用最新的efficientnet，分类数目改成10
注意，该网络改变了两个东西以适应mnistdef load_pretrained_weights(model, model_name): """ Loads pretrained weights, and downloads if loading for the first time. """ state_dict = model_zoo.load_url(url_map[m...原创 2019-06-14 20:50:31 · 3002 阅读 · 6 评论 -
pytorch的各种版本下载-以及pytorch的参数可视化
https://download.pytorch.org/whl/cu100/torch_stable.htmlVia pipDownload thewhlfile with the desired version via this command (you can replace 1.0.1 with the version you choose):export CUDA...原创 2019-06-12 10:46:04 · 7962 阅读 · 4 评论 -
pytorch知识点--[ pytorch ] ——基本使用：(3) finetune冻结层操作 + 学习率超参数设
原[ Pytorch ] —— 复现积累原[ pytorch ] ——基本使用：(8) 优化器optimizer的使用原【python】 —— 库：visdom（pytorch中的可视化工具）原[ pytorch ] ——基本使用：(7) GPU分配原[ pytorch ] ——基本使用：(6) 模型并联转[ pytorch ] ——基本使用：(5) 计算模型参数量...转载 2019-08-14 19:51:19 · 473 阅读 · 0 评论 -
github开源项目pytorch-classification，升级到pytorch1.1.0以上
刚开始就全屏：train_list.txttrain/n01532829/n0153282900000005.jpg 0train/n01532829/n0153282900000006.jpg 0train/n01532829/n0153282900000007.jpg 0train/n01532829/n0153282900000010.jpg 0train/n01...原创 2019-08-14 15:04:26 · 1376 阅读 · 2 评论 -
PyTorch项目应用实例(转载)
PyTorch项目应用实例(八)固定权重|顺序训练网络原PyTorch项目应用实例(七)模型添加中继loss | 中继监督优化原PyTorch项目应用实例(六)并行化|分组运算|张量乘|常用神经网络层原PyTorch项目应用实例(五)加载模型验证并将所有结果写入文件原coco再分组与网络按照分组进行训练原PyTorch项目应用实例(四)设置learning_rate的...转载 2019-08-14 14:35:11 · 3686 阅读 · 0 评论 -
pytorch遇到的一些关键字记录-问题记录
RuntimeError: Failed to export an ONNX attribute, since it's not constant, please try to make things (e.g., kernel size) static if possibleTraceback (most recent call last): File "/home/spple/...原创 2019-08-13 14:05:25 · 2433 阅读 · 3 评论 -
pytorch0.3.0的train代码
import torchvisioncv_version = torchvision.__version__import torchversion = torch.__version__version_m = version.split('.')改了一些东西，具体表现在如何把基于Pytorch 0.3.0的代码改成适用于0.4.0的？https://www.zhihu...原创 2019-08-12 19:04:40 · 607 阅读 · 0 评论 -
pytorch1.1.0和pytorch0.3.0的inference测试流程
PyTorch 0.4新版本 升级指南 no_gradhttps://blog.youkuaiyun.com/jacke121/article/details/80597759import torchvisioncv_version = torchvision.__version__import torchversion = torch.__version__version_m ...原创 2019-08-10 12:14:57 · 1283 阅读 · 0 评论 -
优化PyTorch训练代码
https://sagivtech.com/2017/09/19/optimizing-pytorch-training-code/转载 2019-08-10 11:44:09 · 243 阅读 · 0 评论 -
pytorch0.3.0和pytorch现代框架的1.1.0(0.4.1)的区别
官方文档介绍https://github.com/pytorch/pytorch/releases/tag/v0.4.0最好的介绍文章PyTorch 0.4新版本 升级指南 no_gradhttps://blog.youkuaiyun.com/jacke121/article/details/80597759pytorch0.3和0.4对比总结https://blog.csdn.n...转载 2019-08-10 11:13:38 · 2157 阅读 · 0 评论 -
pytorch模型保存方式-时间测试-inference--torch.squeeze() 和torch.unsqueeze()的相关用法
pytorch中cp27m与cp27mu的区别https://blog.youkuaiyun.com/dlhlSC/article/details/88533376我们在下载pytorch安装包时，会发现CUDA9.0有py27_gpu版有两个whl下载文件，分别是：cu90/torch-0.3.0.post4-cp27-cp27mu-linux_x86_64.whlcu90/torch-0....转载 2019-08-09 14:58:22 · 1500 阅读 · 0 评论 -
person-reid-triplet-loss-baseline-项目分析
关于项目配置：原文中是：I use Python 2.7 and Pytorch 0.3. For installing Pytorch, follow the official guide. Other packages are specified in requirements.txt.opencv_python==3.2.0.7numpy==1.11.3scipy==0.1...原创 2019-08-07 18:27:29 · 433 阅读 · 0 评论