深度学习
zhanghm1995
香港中文大学(深圳)在读博士生
展开
-
解决Pytorch中调用tensorboard报错 AttributeError: module ‘tensorflow‘ has no attribute ‘gfile‘
问题描述在执行下面命令时出现报错:from torch.utils.tensorboard import SummaryWritertb_writer = SummaryWriter(osp.join(opt.savepath, "logdir"))报错内容:AttributeError: module 'tensorflow._api.v1.io' has no attribute 'gfile'问题分析出现这个问题的根本原因在于pytorch调了Tensorflow,最后由Tenso原创 2022-01-09 15:05:00 · 2451 阅读 · 1 评论 -
使用mxnet库报错以及解决方案
错误1 Compile with USE_CUDA=1 to enable GPU usage问题原因:一般是因为你在程序中使用了cuda,但是实际上安装了不是cuda版本的mxnet库。解决方法:你需要看下你安装的mxnet库是否是cuda版本的:pip list | grep mxnet 如果不是cuda版本的,就需要重新安装下cuda版本的mxnet,需要先卸载原有的,并安装对应cuda版本的mxnet,例如你的环境中cuda是10.1的,你就安装mxnet-cu101,类似这样:#原创 2021-08-27 09:51:38 · 4003 阅读 · 0 评论 -
Pytorch学习之旅——tensor张量使用基础
文章目录tensor定义tensor运算tensor内置函数squeeze()和unsqueeze()函数tensor定义创建指定大小的tensorimport torcha=torch.Tensor(2,3)print(a)tensor运算tensor内置函数只写常用的,其他的可以去网站查阅:squeeze()和unsqueeze()函数https://blog.csdn.net/ZT0518/article/details/86440338...原创 2020-08-23 09:26:23 · 601 阅读 · 0 评论 -
Ubuntu18.04 RTX2070 显卡驱动、Cuda、cudnn和Pytorch深度学习环境配置——亲测可用
这篇博文的主要目的是记录我在Ubuntu18.04下配置深度学习环境的过程,方便自己以后配置的需求,也供大家参考,减少跳坑电脑环境说明:系统:Ubuntu18.04 LTS显卡:RTX 2070 Supercuda:10.1cudnn:7.6.51、依赖项安装$ sudo apt-get update$ sudo apt-get upgrade$ sudo apt-ge...原创 2020-04-06 11:51:22 · 7020 阅读 · 8 评论 -
深度学习框架GTX960+cuda9.0+cudnn安装
在ubuntu下使用深度学习框架,不可避免要用到GPU加速,也不可避免会使用cuda和cudnn两个库。本篇博文主要记录自己在配置caffe深度学习框架时安装cuda和cudnn的过程电脑配置:ubuntu16.04 + Nvdia GTX960M现在主流caffe框架安装的都是cuda8.0的版本,cuda官网上已经有9.0的版本,大致了解了一下,9.0的版本也比较适合我当前配置使用,因此选择安...原创 2018-04-11 22:27:34 · 12263 阅读 · 2 评论 -
darknet_ros源码解析(一)-代码配置
YOLO(You Only Look Once) 作为一个检测速度非常快,检测目标种类极其丰富的深度学习算法,受到了非常多的研究人员和深度学习爱好者的青睐。网上有关darknet(YOLO算法的框架)的代码已经有非常详细的博客介绍,但因为平时喜欢在ROS框架下进行算法开发,便于做一些实际应用,所以就想要在能够在ROS框架下使用YOLO,所幸已经有大牛将YOLO写成了非常好用ROS包——darkn...原创 2018-11-22 20:19:46 · 4878 阅读 · 27 评论