基于pymesh镜像Docker构建时无法安装torch==1.9.1+cu102如何解决

报错根因 #

带+cu102后缀的PyTorch版本是适配CUDA10.2的专属构建版本，没有发布到PyPI官方源，直接使用默认源执行pip安装无法找到对应安装包。

可选择的解决方法 #

• 方法一：指定PyTorch专属源安装（无需修改requirements.txt）
  修改Dockerfile中安装依赖的命令，添加PyTorch官方源地址参数：

  RUN pip install -r requirements.txt -f https://download.pytorch.org/whl/torch_stable.html
  

  同时可以结合报错提示的pip版本升级操作，避免旧版本pip的兼容问题：

  RUN /usr/local/bin/python -m pip install --upgrade pip \
      && pip install -r requirements.txt -f https://download.pytorch.org/whl/torch_stable.html
  

• 方法二：调整依赖版本（适合不需要GPU加速的场景）
  直接修改requirements.txt中的对应行，将torch==1.9.1+cu102替换为torch==1.9.1，即可从PyPI默认源拉取对应版本的PyTorch安装包。

额外注意事项 #

如果你的应用需要使用GPU加速，需要确认基础镜像pymesh/pymesh:latest内置的CUDA版本和你安装的PyTorch CUDA版本匹配，否则即使安装成功也无法正常调用GPU资源。

内容的提问来源于stack exchange，提问作者Aman Savaria