m0_53905657的博客

原创 在ARM架构下CentOS9部署CUDA与pytorch

在ARM架构下部署CUDA与pytorch有许多坑，要多研究多想，找官方文档。本人在课程中需要服务器进行训练，由于学校给的所里服务器为单卡V100，课程300人共用，时常抢不到算力。为了能独占一个显卡，自己搭设了一台单卡A100服务器，用于训练。但是搭设的设备为ARM架构，并不是x86架构，并没有现成的pytorch可以用，（其实有，但是不能使用CUDA计算），以下使用官方镜像，最后成功部署。系统终端输入uname --all，可以看到系统架构。本人使用的是aarch64，发行版为CentOS9。