pytoch
文章平均质量分 58
左&&右
这个作者很懒,什么都没留下…
展开
-
创建自己的图像分割数据集并利用Unet实现图像分割(pytorch)
一、 配置lanelme数据集标注工具1.安装labelme二、实现数据标注后图像分割建立数据集三、利用FCN实现数据分割原创 2021-12-08 16:22:46 · 9240 阅读 · 2 评论 -
pytorch使用多个GPU分别独立训练的方式
GPU的编号一般默认从0开始依次递增,利用代码os.environ["CUDA_VISIBLE_DEVICES"] = '1'可以指定你想用的GPU。当你想指定多个的时候可以用os.environ["CUDA_VISIBLE_DEVICES"] = '1,2,3'注意,修改指定GPU后device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")此代码块中的0不用改变。...原创 2021-10-14 11:11:11 · 316 阅读 · 0 评论 -
pytorch显存不足时的解决办法
将float32转化为float16,是最有效的降低显存占用的方式,可以降低一半左右的显存占用。实现方式:首先在代码的最前面加上torch.set_default_dtype(torch.float16)这行代码将这个程序内部所有的float变量转化为float32。此时如果直接运行程序会出现输入为float参数为floathalf的错误报告。然后将输入改变为floathalf即可代码如下 inputs = inputs.type(torch.float16)...原创 2021-10-14 11:05:52 · 7813 阅读 · 0 评论 -
pytorch安装以及pycharm使用pytorch包(windows10)
1. 安装anacondahttps://www.anaconda.com/products/individual#windows在该网站上下载anaconda的包,然后使用默认设置安装即可。为anaconda配置环境变量anaconda和miniconda同为conda官方推荐的conda工具。anaconda和conda的关系,就像github和git的关系,anaconda不光包含了conda命令行软件,还包含了多如牛毛的预装python包,以及令人瞠目结舌的黑科技全家桶。win+R打开原创 2021-04-01 09:06:13 · 1813 阅读 · 2 评论