
Deep Learning
baiyuwujia
这个作者很懒,什么都没留下…
展开
-
Ubuntu系统 nvidia-docker创建深度学习容器
任务:创建新的container,镜像使用ubuntu系统Step 1 : 拉取ubuntu镜像docker search ubuntu # 查找镜像资源docker pull ubuntu # 拉取镜像这个ubuntu基本上就是个空架子,里面啥都没,全都需要自己重新安装Step 2 : 创建容器1、查看现有的容器和镜像''' 查看所有container '''...原创 2020-05-18 09:57:49 · 1434 阅读 · 0 评论 -
nvcc: command not found
nvidia-smi可用,但是nvcc不可用,这种问题可能是以下两种情况:1、nvcc命令没有添加到系统路径https://blog.youkuaiyun.com/Ocean_waver/article/details/102730467?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.nonecase&depth_1-utm_source=distribute.pc_relevant.原创 2020-06-10 18:58:03 · 3297 阅读 · 0 评论 -
复现CenterNet
源码地址:https://github.com/xingyizhou/CenterNet新手上路先看readme.md,本文按照readme的思路一步步复现1.InstallationPlease refer toINSTALL.mdfor installation instructions.(这部分有些git clone太慢了,我都是直接下载下来然后用Xftp传到服务器上的)...原创 2019-10-10 14:07:29 · 2077 阅读 · 1 评论 -
No module named ''; '' is not a pack
包名和包中文件名相同,此时在包内部运行代码就会出错原创 2019-08-26 11:54:11 · 319 阅读 · 0 评论 -
pytorch使用gpu(linux服务器上)
1.查看服务器中GPU信息: 转自这里-->https://blog.youkuaiyun.com/handsome_bear/article/details/80903477 nvidia-smi2.使用单个gpu: 转自-->https://blog.youkuaiyun.com/yzy__zju/article/details/85014576# 指定gpu/cpu, 指定...原创 2019-08-20 14:15:42 · 2299 阅读 · 0 评论 -
tqdm进度条 在pytorch中的使用
引入该模块:from tqdm import tqdmtqdm可用于所有可迭代对象,所以可以直接用在dataloader上,如下:for data, target in tqdm(train_loader): ...效果:tqdm用在dataloader上其实是对每个batch和batch总数做的进度条,我也是在训练比较慢时写出来看看训练到哪了。。。...原创 2019-08-20 10:27:29 · 24601 阅读 · 1 评论 -
pytorch报错: invalid argument 0: Sizes of tensors must match except in dimension 0.
pytorch报错:invalid argument 0: Sizes of tensors must match except in dimension 0. Got 13 and 0 in dimension 2 at /opt/conda/conda-bld/pytorch_1556653114079/work/aten/src/TH/generic/THTensor.cpp:711...原创 2019-08-08 09:44:38 · 2224 阅读 · 0 评论 -
镜像padding一张图片,利用pytorch.nn.functional.pad()
import torch as timport torch.nn as nndef ReflectPad2D(image, pad_size): ''' 镜像padding :param image: 图像 :param pad_size: padding大小,我这里简化成正方形图,四个方向全padding同样的值 :return: padding...原创 2019-07-11 16:50:18 · 1099 阅读 · 0 评论 -
pytorch安装(win10)
官方安装教程:https://pytorch.org/get-started/locally/使用anaconda:开始界面打开Anaconda Prompt然后输入相关命令查看CUDA版本:https://www.jianshu.com/p/d3b9419a0f89...原创 2019-05-29 19:32:50 · 928 阅读 · 0 评论