WSL2 Docker容器中torchrun: command not found问题咨询

一、"torchrun: command not found" 错误修复 #

• 先确认容器内是否安装PyTorch：在容器终端执行 pip list | grep torch，无输出则说明未安装，执行对应版本的安装命令，比如适配CUDA 11.8的命令：pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
• 若已安装PyTorch仍找不到torchrun，检查环境变量：执行 echo $PATH，查看PyTorch的bin目录是否在PATH中。若不在，手动添加（示例路径，需根据实际调整）：export PATH=$PATH:/usr/local/lib/python3.8/site-packages/torch/bin，也可将该命令写入容器的~/.bashrc实现永久生效

二、项目文件与容器的操作逻辑纠正 #

• 先下载项目文件再进入隔离容器的操作不正确。Docker容器默认是隔离环境，本地文件不会自动映射至容器内。正确流程：
  1. 拉取镜像后，启动容器时通过挂载卷将本地项目文件、模型权重目录映射到容器内
  2. 例如在docker run命令中添加 -v /本地项目路径:/容器内目标路径 -v /本地模型权重路径:/容器内模型权重路径 参数，让容器直接访问本地文件

三、容器访问本地模型权重与脚本的解决方法 #

• 修改启动容器的run.sh脚本，在docker run命令中添加挂载参数：

  示例：假设本地项目文件在/home/yourname/project，模型权重在/home/yourname/models，要映射到容器内的/workspace/project和/workspace/models，则run.sh中的命令改为：
  docker run -it --gpus all -v /home/yourname/project:/workspace/project -v /home/yourname/models:/workspace/models 镜像名 bash

• 启动容器后，进入挂载目录（如cd /workspace/project），再执行generate_interactive.sh脚本即可访问本地模型和脚本

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