1条回答 默认 最新
关注
【相关推荐】
- 这个问题的回答你可以参考下: https://ask.csdn.net/questions/7539770
- 我还给你找了一篇非常好的博客,你可以看看是否有帮助,链接:AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘
- 同时,你还可以查看手册:pytorch torch.backends.cuda module 中的内容
- 除此之外, 这篇博客: 关于AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘问题的解决中的 2.解决方法 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
1)修改执行命令
由于我的pytorch装的是CPU版本的,因此需要在python执行命令后面加上 --gpu_ids -1。
E:\PythonCode\Net>python train.py --gpu_ids -1
这样运行就没有出错了,不过,还可以用另一种方式解决~
2)修改"base_options.py"文件
这个文件在错误信息里面可以找到。
def initialize(self, parser): """Define the common options that are used in both training and test.""" # basic parameters parser.add_argument('--dataroot', type=str, default='./LEVIR-CD', help='path to images (should have subfolders A, B, label)') parser.add_argument('--val_dataroot', type=str, default='./LEVIR-CD', help='path to images in the val phase (should have subfolders A, B, label)') parser.add_argument('--name', type=str, default='experiment_name', help='name of the experiment. It decides where to store samples and models') parser.add_argument('--gpu_ids', type=str, default='-1', help='gpu ids: e.g. 0 0,1,2, 0,2. use -1 for CPU') parser.add_argument('--checkpoints_dir', type=str, default='./checkpoints', help='models are saved here') # model parameters parser.add_argument('--model', type=str, default='CDF0', help='chooses which model to use. [CDF0 | CDFA]') parser.add_argument('--input_nc', type=int, default=3, help='# of input image channels: 3 for RGB ') parser.add_argument('--output_nc', type=int, default=3, help='# of output image channels: 3 for RGB') parser.add_argument('--arch', type=str, default='mynet3', help='feature extractor architecture | mynet3') parser.add_argument('--f_c', type=int, default=64, help='feature extractor channel num') parser.add_argument('--n_class', type=int, default=2, help='# of output pred channels: 2 for num of classes') parser.add_argument('--init_type', type=str, default='normal', help='network initialization [normal | xavier | kaiming | orthogonal]') parser.add_argument('--init_gain', type=float, default=0.02, help='scaling factor for normal, xavier and orthogonal.') parser.add_argument('--SA_mode', type=str, default='BAM', help='choose self attention mode for change detection, | ori |1 | 2 |pyramid, ...')
直接把’–gpu_ids’的 default改成’-1’,简单粗暴。(原来默认值是0)
以上,完美解决了问题。
如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^解决 无用评论 打赏 举报
悬赏问题
- ¥200 csgo2的viewmatrix值是否还有别的获取方式
- ¥15 Stable Diffusion,用Ebsynth utility在视频选帧图重绘,第一步报错,蒙版和帧图没法生成,怎么处理啊
- ¥15 请把下列每一行代码完整地读懂并注释出来
- ¥15 pycharm运行main文件,显示没有conda环境
- ¥15 易优eyoucms关于二级栏目调用的问题
- ¥15 寻找公式识别开发,自动识别整页文档、图像公式的软件
- ¥15 为什么eclipse不能再下载了?
- ¥15 编辑cmake lists 明明写了project项目名,但是还是报错怎么回事
- ¥15 关于#计算机视觉#的问题:求一份高质量桥梁多病害数据集
- ¥15 特定网页无法访问,已排除网页问题