python - AzureML Notebook AssertionError：Torch 未在启用 CUDA 的情况下编译

Question

尝试在 azureML notebook 中执行此代码块时出现上述错误：

cfg = get_cfg()
cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml"))
cfg.MODEL.weights = model_zoo.get_checkpoint_url("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml")
predictor = DefaultPredictor(cfg)

有谁知道为什么？顺便说一句，我正在使用的 conda 环境是这样的：

channels:
  - conda-forge
  - pytorch
  - anaconda
  - defaults
dependencies:
  - python==3.8
  - matplotlib
  - numpy
  - pytorch==1.8.0
  - torchvision==0.9.0
  - cpuonly
  - scikit-learn
  - pip
  - pip:
      - azureml-defaults
      - pandas
      - joblib
      - opencv-python-headless
      - 'git+https://github.com/facebookresearch/detectron2.git'
name: azureml_350ac2b8c050124461fa60260e56928b

我先谢谢你！

Answer 1

谢谢你。发布您的建议作为帮助其他社区成员的答案。

因为该cpuonly软件包禁用了 PyTorch 的 GPU（即 CUDA）构建。将 替换cpuonly为您的 CUDA 工具包版本的规范，例如。cudatoolkit=10.2. 此外，pytorch通道应该几乎总是在 conda - forge通道之上

可以参考AssertionError: Torch not compiled with CUDA enabled