【cuda】cuda安装，以及配置mindspore

jianyunchao commented 1 year ago

gpu：通用计算芯片-硬件 cuda driver：显卡驱动，识别硬件并交互-底层接口 cuda toolkit：开发工具，并行计算框架-应用程序接口

jianyunchao commented 1 year ago

cuda显卡驱动

nvidia官网显卡驱动下载： https://www.nvidia.cn/Download/index.aspx?lang=cn

根据不同的gpu显卡芯片下载对应驱动的run包安装命令： bash xxx.run

jianyunchao commented 1 year ago

cuda工具包安装

官网快速下载：https://developer.nvidia.com/cuda-downloads 官网版本汇总：https://developer.nvidia.com/cuda-toolkit-archive

安装： ./cuda_11.6.0_510.39.01_linux.run = Summary = Driver: Installed Toolkit: Installed in /usr/local/cuda-11.6/ Please make sure that

PATH includes /usr/local/cuda-11.6/bin
LD_LIBRARY_PATH includes /usr/local/cuda-11.6/lib64, or, add /usr/local/cuda-11.6/lib64 to /etc/ld.so.conf and run ldconfig as root To uninstall the CUDA Toolkit, run cuda-uninstaller in /usr/local/cuda-11.6/bin To uninstall the NVIDIA Driver, run nvidia-uninstall Logfile is /var/log/cuda-installer.log

accept 选择driver和其他下一步

jianyunchao commented 1 year ago

安装cuDNN

下载地址： https://developer.nvidia.cn/rdp/cudnn-archive

tar -xf cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz cd cudnn-linux-x86_64-8.5.0.96_cuda11-archive sudo cp include/cudnn.h /usr/local/cuda-11.6/include sudo cp lib/libcudnn /usr/local/cuda-11.6/lib64 sudo chmod a+r /usr/local/cuda-11.6/include/cudnn.h /usr/local/cuda-11.6/lib64/libcudnn

jianyunchao commented 1 year ago

环境变量设置

export PATH=/usr/local/cuda-11.6/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64:$LD_LIBRARY_PATH

jianyunchao / mindlearning