获得当前环境

通过下载并运行PyTorch官方提供的collect_env.py脚本,可以快速获取当前系统的PyTorch环境配置详情,包括CUDA版本、Python版本、依赖库等信息,便于排查环境问题。

作者:zhuge··预计阅读 1 分钟·631 阅读·0 评论
获得当前环境
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

相关文章

评论

加载中...