ChrisKim
Do not go gentle into that good night.
颢天

机器学习
文章归档

使用容器在国产超算平台安装 PyTorch CUDA

近期知道了我校新建了高性能计算平台,于是找老师申请了个账号。上去一看发现居然三台 GPU 节点,每台都有两块 A100 40G,遂打算开始炼丹。结果发现学校为了国产化,服务器 CPU 居然是 arm64 指令集的鲲鹏 920,导致 PyTorch 无法安装。折腾许久,总结了本篇文章,用容…

   2024-03-11   2   查看全文