银河麒麟V10P3系统+ARM64架构下,手动编译pytorch,适配T4卡算力

下载所需的NVIDIA驱动、cuda、cudnn、anaconda、pytorch源码

NVIDIA驱动下载安装

在银河麒麟系统中禁掉系统自带的nouveau显卡驱动,不同的系统blacklist.conf文件在不同的位置,在终端命令窗口输入以下命令进入相应的文件:

sudo vim /etc/modprobe.d/blacklist.conf

sudo vim /usr/lib/modprobe.d/dist-blacklist.conf

加入blacklist nouveau禁用驱动程序。
注释#blacklist nvidiafb(查找文件内没有,我还是老老实实的写入了#blacklist nvidiafb,怕后续会有问题,虽然注释了就跟没有一样)

sudo dracut --force
sudo reboot

(拔掉显示器。)
安装下载好的驱动

sudo ./NVIDIA-Linux-aarch64-525.147.05.run  -no-x-check -no-nouveau-check -no-opengl-files 

下载安装cuda118

下载地址https://developer.nvidia.com/cuda-11-8-0-download-archive
图片[1] - 银河麒麟V10P3系统+ARM64架构下,手动编译pytorch,适配T4卡算力 - 宋马
在线下载并且安装

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux_sbsa.run
sudo sh cuda_11.8.0_520.61.05_linux_sbsa.run

配置环境变量

sudo vim ~/.bashrc
export CUDA_HOME=/usr/local/cuda-11.8
export PATH=$PATH:$
© 版权声明
THE END
如果内容对您有所帮助,就支持一下吧!
点赞0 分享
评论 抢沙发

请登录后发表评论

    暂无评论内容