torch-1.14.0+cuda11.8+sm-90+deepspeed应对新一代nvidia H100卡的高算力需求
经过编译的torch-1.14,支持cuda11.8和nvidia H100卡的算力需求达到90,可以在deepspeed环境下训练大模型。兼容环境包括ubuntu:22.04,python:3.10.6,nvidia-smi:525.105.17,cuda:11.8,deepspeed:0.9.2,cpu:intel。需要其他算力支持请与我联系。
下载地址
用户评论
包是好包,版本和我的不匹配,感谢