【Python】torchrun 命令执行环境修改为 conda环境

最新推荐文章于 2025-04-23 14:14:45 发布

原小明

最新推荐文章于 2025-04-23 14:14:45 发布

阅读量488

点赞数 2

文章标签： python conda 开发语言

本文链接：https://blog.csdn.net/LABLENET/article/details/143887216

版权

1，查看真实路径

conda activate env

输出：

$ which python
/home/metax/miniconda3/envs/cnn/bin/python

2，编辑 torchrun

$ vim ~/.local/bin/torchrun
#!/home/metax/miniconda3/envs/cnn/bin/python 这里修改为上述路径
# -*- coding: utf-8 -*-
import re
import sys
from torch.distributed.run import main
if __name__ == '__main__':
    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
    sys.exit(main())

保存，授权，chmod +x ~/.local/bin/torchrun

执行torchrun -h 验证