我想在 brainframe 中使用 Pytorch，所以在本地把 pytorch 安装到了 aotuai/brainframe_core:0.28.3 容器中，但出现了一些错误。有人能给点建议吗？

你好！这是个好问题——我不确定你的图片里发生了什么，不过这里有一个快速教程，介绍如何修改 BrainFrame 镜像里的 python 库。希望对你有帮助！

构建说明

1. 首先，下载下面的 Dockerfile，并 cd 到同一目录。
   Dockerfile (132 Bytes)

2. 为你要使用的 brainframe 版本设置一个环境变量：

   export BRAINFRAME_VERSION=0.28.3
   

3. 在 Dockerfile 所在的目录中，使用以下命令运行构建：

   docker build \
       --build-arg BRAINFRAME_VERSION=$BRAINFRAME_VERSION \
       --build-arg PYTHON_LIBRARIES=torch \
       --tag aotuai/brainframe_core:$BRAINFRAME_VERSION \
       .
   

要更改想要安装的库，只需修改上面 docker build 命令中的 PYTHON_LIBRARIES 变量。

然后，你可以运行下面的命令来测试新镜像是否有效：

docker run -it aotuai/brainframe_core:0.28.3 python3 -c "import torch;print(torch.__version__)"

它会返回容器内当前安装的 pytorch 版本。

谢谢，Alex！我按你的建议安装了 PyTorch，但仍然有一些错误。你能给点建议吗？

看起来你的包名拼写有误。我想你要的包是“torch”。

谢谢！这个错误真尴尬。修改之后我仍然有错误。d509042f677ebd8a7d9c3fe63652ad7|336x500

我已经在服务端上成功安装了 PyTorch。可能只是虚拟机安装方面存在一些小问题。不管怎样，谢谢你。