AttributeError: module ‘torch.onnx‘ has no attribute ‘set_training‘

原创于 2021-06-23 10:23:13 发布

· 3.1k 阅读

3 ·

版权

文章标签：

#bug #pytorch #深度学习 #神经网络

深度学习同时被 3 个专栏收录

34 篇文章

订阅专栏

Bug日常

10 篇文章

订阅专栏

Pytorch

3 篇文章

订阅专栏

由于Pytorch 1.6以上版本移除了'torch.onnx'中的'set_training'属性，导致在1.8版本中使用tensorwatch绘制ResNet18模型结构时出现错误。解决方法是将Pytorch版本降至1.6以下，如1.5.1，并提供了针对不同CUDA版本和CPU的安装命令。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述

import torch
import tensorwatch as tw
resnet18 = models.resnet18(pretrained=True)
tw.draw_model(resnet18, (1, 3, 224, 224))

使用tensorwatch展示网络结构时报错。

问题原因

当前环境中Pytorch版本为1.8。
高于1.6的版本中，模块’torch.onnx’去掉了 'set_training’这个属性。

解决办法

将Pytorch降到1.6以下。

Conda 安装 v1.5.1

# CUDA 9.2
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=9.2 -c pytorch

# CUDA 10.1
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.1 -c pytorch

# CUDA 10.2
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.2 -c pytorch

# CPU Only
conda install pytorch==1.5.1 torchvision==0.6.1 cpuonly -c pytorch

pip 安装 v1.5.1

# CUDA 10.2
pip install torch==1.5.1 torchvision==0.6.1

# CUDA 10.1
pip install torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 9.2
pip install torch==1.5.1+cu92 torchvision==0.6.1+cu92 -f https://download.pytorch.org/whl/torch_stable.html

# CPU only
pip install torch==1.5.1+cpu torchvision==0.6.1+cpu -f https://download.pytorch.org/whl/torch_stable.html