Question

我正在尝试使用Colab平台来完成关于CycleGANs的教程，但是我正在努力寻找一种方法来解决仅导入库的“简单”部分。

我只是想导入以下内容：

from fastai.conv_learner import *
from fastai.dataset import *
from cgan.options.train_options import *
from sklearn.model_selection import train_test_split
from cgan.options.train_options import TrainOptions
from cgan.data.data_loader import CreateDataLoader
from cgan.models.models import create_model
from cgan.util.visualizer import Visualizer
from google_images_download import google_images_download

我目前在第一行遇到错误：

----> 7 class IntermediateLayerGetter(nn.ModuleDict):
      8     """
      9     Module wrapper that returns intermediate layers from a model

AttributeError: module 'torch.nn' has no attribute 'ModuleDict'

可以在网上找到代码：https://colab.research.google.com/drive/1dMQWStzLfAHDTGuKaUNQn1aOBWeJw4FN

Answer 1

您是否选中了此issue，它可以部分解决问题。

此后，我收到此错误（检测到PyTorch和torchvision是使用不同的CUDA版本编译的。PyTorch的CUDA版本= 9.0，而torchvision的CUDA版本= 10.0。请重新安装与您的PyTorch安装匹配的torchvision。）

pip install torch==1.0.1 -f https://download.pytorch.org/whl/cu100/stable

我希望这会有所帮助：）

在Colab上设置FastAi时出现问题

1 个答案: