本期目录
Linux安装CUDA
- 输入以下命令,查看 GPU 支持的最高 CUDA 版本。笔者这里显示的是 11.6 ,这意味着,安装的 CUDA 版本必须 <= 11.6 。
nvidia-smi
- 前往 Nvidia 的 CUDA 官网:CUDA Toolkit Archive | NVIDIA Developer ,笔者下载 CUDA 11.6.2 版本:
- 如下图选择:
- 选择好后,下方会显示两条 Linux 命令。在 Linux 终端输入这条命令:
$ wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
- 这里笔者的服务器无法翻墙,直接在服务器上下载会超时。所以直接复制下载地址,先在 Windows 系统下下载好,再上传到
/root/
目录下。https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
- 这里由于笔者没有 root 权限,无法上传到
/root/
目录下。因此将此安装包上传至/home/og/xxx
自己的工作目录下。输入以下命令:$ sudosh cuda_11.6.2_510.47.03_linux.run
- 弹出来第一个选择框,笔者这台服务器已经安装过 Nvidia 的显卡驱动了,因此选择 “continute” ,会出现下面这个对话框,选择 “accept” 。
- 笔者服务器已经安装过 Nvidia 显卡驱动了,因此不需要再安装了。如果你尚未安装驱动,可以顺便一起安装了。摁一下空格取消 Driver 安装,直接选择 Install 安装:
- 安装成功:
- 接下来配置环境变量,直接在 Linux 命令行输入以下命令:
sudovim ~/.bashrc
在文件末尾添加 (这里笔者假设你已经熟悉 Linux 下 Vim 文本编辑器的使用,不熟悉的强烈推荐去学习一下) :exportPATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}}exportLD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}exportCUDA_HOME=/usr/local/cuda-11.6
- 更新系统环境:
source ~/.bashrcsudo ldconfig
- 验证 CUDA 是否安装成功,输入以下命令:
nvcc -V
输出下图即为成功安装:
本文转载自: https://blog.csdn.net/Sihang_Xie/article/details/127347139
版权归原作者 自牧君 所有, 如有侵权,请联系我们删除。
版权归原作者 自牧君 所有, 如有侵权,请联系我们删除。