wget https://raw.githubusercontent.com/pytorch/pytorch/main/torch/utils/collect_env.py
# For security purposes, please check the contents of collect_env.py before running it.
python collect_env.py
获得当前环境
通过下载并运行PyTorch官方提供的collect_env.py脚本,可以快速获取当前系统的PyTorch环境配置详情,包括CUDA版本、Python版本、依赖库等信息,便于排查环境问题。
作者:zhuge··预计阅读 1 分钟·631 阅读·0 评论

评论