ubuntu cuda10.1

### 如何在 Ubuntu 上安装和配置 CUDA 10.1 #### 准备工作 为了确保顺利安装 CUDA 10.1，在开始之前需要确认系统的环境满足最低需求。这包括检查操作系统版本、驱动程序兼容性和 GCC 版本。 - **验证 NVIDIA 显卡驱动** NVIDIA 的显卡驱动是运行 CUDA 所需的核心组件之一。可以通过以下命令来检测当前已安装的驱动版本： ```bash nvidia-smi ``` 如果未显示任何信息，则可能尚未安装合适的 NVIDIA 驱动程序，可以按照官方文档中的说明进行安装[^1]。 - **调整 GCC 版本** 某些较新的 Ubuntu 发行版可能会默认使用更高版本的 GCC 编译器，而这些高版本可能导致与 CUDA 工具链不兼容的情况发生。因此建议将 GCC 升级或者降级至适合 CUDA 10.1 使用的稳定版本 (通常是 GCC 7.x 或者更低)[^3]。 #### 开始安装过程 以下是具体的操作指南： - **添加 NVIDIA 软件仓库并更新索引** 打开终端输入如下指令以引入最新的 NVIDIA 存储库列表文件以及同步本地缓存数据： ```bash sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt-get update ``` - **下载并执行 CUDA 安装脚本** 前往[NVIDIA 官方网站](https://developer.nvidia.com/cuda-downloads)，找到对应于目标平台架构下的 CUDA Toolkit 下载链接地址；接着利用 `wget` 获取二进制镜像包，并赋予可执行权限后启动交互式向导模式完成整个部署流程[^4]: ```bash cd ~/Downloads/ wget https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.168_418.67_linux.run chmod +x cuda_10.1.168_418.67_linux.run sudo ./cuda_10.1.168_418.67_linux.run --override ``` 注意这里附加参数 `--override` 是强制忽略部分潜在冲突警告继续前进的方式。 - **设置必要的环境变量** 编辑用户的 shell profile 文件（比如 `.bashrc`,`.zshrc`) 来追加路径定义语句以便后续调用编译工具时能够自动定位到正确位置: ```bash echo 'export PATH=/usr/local/cuda-10.1/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc source ~/.bashrc ``` 最后重启计算机使更改生效即可测试新安装好的 CUDA 是否正常运作。 --- ### 测试 CUDA 功能是否可用 完成上述全部操作之后，可通过下面这段简单的 C++ 程序片段来进行功能性的初步检验: ```cpp #include <stdio.h> int main() { int nDevices; cudaGetDeviceCount(&nDevices); printf("Number of devices: %d\n", nDevices); for(int i=0;i<nDevices;i++) { cudaDeviceProp prop; cudaGetDeviceProperties(&prop,i); printf("\n Device Number:%d \n",i); printf(" Device name:%s\n",prop.name); printf(" Memory Clock Rate(KHz):%d\n",prop.memoryClockRate); printf(" Memory Bus Width(bits):%d\n",prop.memoryBusWidth); printf(" Peak Memory Bandwidth(GB/s):%.2f\n", 2.0*prop.memoryClockRate*(prop.memoryBusWidth/8)/1e6); } } ``` 保存为 test.cu 并构建运行它查看输出结果是否符合预期。 ---