关于 pytorch：如何从 Google Colab 运行 lua ML 模型


How to run lua ML model from Google Colab

我正在尝试从 Github 但在 Google Colab 上运行这种神经风格迁移模型，因为我的计算机没有足够的内存/CPU/任何东西。

我已经将我的谷歌驱动器安装到我的笔记本上，按照本教程将模型下载到我的驱动器文件夹中，将存储库克隆到我的驱动器上，为了测试它是否有效，我正在使用运行的最基本示例自述文件中的 Brad Pitt 示例使用：

1
th neural_style.lua -style_image examples/inputs/picasso_selfport1907.jpg -content_image examples/inputs/brad_pitt.jpg

但由于某种原因它不起作用。

  • 使用 subprocess.run() 刚刚返回 No such file or directory
  • 使用 !th neural_style.lua... 返回 th: command not found

    • 我已经尝试了其他四件事，它们都给了我上述两个错误消息的变体。有什么想法吗？

    这是完整的笔记本代码，用于从头到尾重现我在 Colab 上的设置/错误：

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    # Mount the drive
    from google.colab import drive
    drive.mount('/content/drive')

    # Clone the repo onto the drive
    !git clone https://github.com/jcjohnson/neural-style

    # Install Pytorch
    !pip3 install http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl
    !pip3 install torchvision

    # Download the models per the github repo instructions
    !bash models/download_models.sh

    !th neural_style.lua -style_image examples/inputs/picasso_selfport1907.jpg -content_image examples/inputs/brad_pitt.jpg

    第一：

    1
    2
    3
    4
    5
    6
    7
    %cd /content/
    !git clone https://github.com/nagadomi/distro.git torch --recursive
    import os
    os.chdir('./torch/')
    !bash install-deps
    !./install.sh
    !. ./install/bin/torch-activate

    现在应该可以使用 th:

    的绝对路径

    1
    !/content/torch/install/bin/th neural_style.lua -style_image examples/inputs/picasso_selfport1907.jpg -content_image examples/inputs/brad_pitt.jpg