安装NVIDIA CUDA组件
操作步骤
- 使用PuTTY工具,以root用户登录服务器。
- 执行以下命令获取NVIDIA CUDA组件。
wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux_sbsa.run
- 执行以下命令安装NVIDIA CUDA组件。
sudo shcuda_11.4.0_470.42.01_linux_sbsa.run
- 执行以下命令进行验证。
ls -l /usr/local/cuda-11.4/
- 执行以下命令配置CUDA的环境变量。
export CUDA_HOME=/usr/local/cuda-11.4 export PATH=/usr/local/cuda-11.4/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH
父主题: 配置编译环境